wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-30 13:47:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

revert omp threads fix (#7288)

Browse Source
This commit is contained in:
Eric Liang
2020-02-23 21:26:49 -08:00
committed by GitHub
parent e7d0ec9531
commit 0ae4fe020d
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/__init__.py
+6
View File
@@ -13,6 +13,12 @@ if "pickle5" in sys.modules:
"requires a specific version of pickle5 (which is "
"packaged along with Ray).")
if "OMP_NUM_THREADS" not in os.environ:
logger.debug("[ray] Forcing OMP_NUM_THREADS=1 to avoid performance "
"degradation with many workers (issue #6998). You can "
"override this by explicitly setting OMP_NUM_THREADS.")
os.environ["OMP_NUM_THREADS"] = "1"
# Add the directory containing pickle5 to the Python path so that we find the
# pickle5 version packaged with ray and not a pre-existing pickle5.
pickle5_path = os.path.join(