Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!19859 preload gomp when import mindspore 

Merge pull request !19859 from ms_yan/r1.3_ctype_preload
This commit is contained in:
i-robot 2021-07-10 02:38:59 +00:00 committed by Gitee
parent c72bb20727 83522df058
commit 104782f257
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/run_check/_check_version.py
View File

@ -392,6 +392,14 @@ def check_version_and_env_config():
os.environ["MS_CLOSE_VERSION_CHECK"] = "ON" os.environ["MS_CLOSE_VERSION_CHECK"] = "ON"
if __package_name__.lower() == "mindspore-ascend": if __package_name__.lower() == "mindspore-ascend":
env_checker = AscendEnvChecker() env_checker = AscendEnvChecker()
# Note: pre-load libgomp.so to solve error like "cannot allocate memory in statis TLS block"
try:
import ctypes
ctypes.cdll.LoadLibrary("libgomp.so.1")
except OSError:
logger.warning(
"Pre-Load Lirary libgomp.so.1 failed, this might cause cannot allocate TLS memory problem, "
"if so find solution in FAQ in https://www.mindspore.cn/doc/faq/en/master/index.html.")
elif __package_name__.lower() == "mindspore-gpu": elif __package_name__.lower() == "mindspore-gpu":
env_checker = GPUEnvChecker() env_checker = GPUEnvChecker()
else: else: