wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-09 19:41:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change log level for OMP warning (#7114)

Browse Source
This commit is contained in:
Simon Mo
2020-02-11 14:15:38 -08:00
committed by GitHub
parent 2a0e4d94aa
commit 039d2cde88
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/source/configure.rst
+8
View File
@@ -48,6 +48,14 @@ If using the command line, connect to the Ray cluster as follow:
# Connect to ray. Notice if connected to existing cluster, you don't specify resources.
ray.init(address=<address>)
.. note::
Ray sets the environment variable ``OMP_NUM_THREADS=1`` by default. This is done
to avoid performance degradation with many workers (issue #6998). You can
override this by explicitly setting ``OMP_NUM_THREADS``. ``OMP_NUM_THREADS`` is commonly
used in numpy, PyTorch, and Tensorflow to perform multit-threaded linear algebra.
In multi-worker setting, we want one thread per worker instead of many threads
per worker to avoid contention.
Logging and Debugging
---------------------
python/ray/__init__.py
+3 -3
View File
@@ -14,9 +14,9 @@ if "pickle5" in sys.modules:
"packaged along with Ray).")
if "OMP_NUM_THREADS" not in os.environ:
logger.warning("[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.")
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