|
|
|
@ -190,15 +190,21 @@ class AscendEnvChecker(EnvChecker):
|
|
|
|
|
"match, reference to the match info on: https://www.mindspore.cn/install")
|
|
|
|
|
|
|
|
|
|
def check_deps_version(self):
|
|
|
|
|
# in order to update the change of 'LD_LIBRARY_PATH' env, run a sub process
|
|
|
|
|
"""
|
|
|
|
|
te, topi, hccl wheel package version check
|
|
|
|
|
in order to update the change of 'LD_LIBRARY_PATH' env, run a sub process
|
|
|
|
|
"""
|
|
|
|
|
input_args = ["--mindspore_version=" + __version__]
|
|
|
|
|
for v in self.version:
|
|
|
|
|
input_args.append("--supported_version=" + v)
|
|
|
|
|
deps_version_checker = os.path.join(os.path.split(os.path.realpath(__file__))[0], "_check_deps_version.py")
|
|
|
|
|
call_cmd = [sys.executable, deps_version_checker] + input_args
|
|
|
|
|
process = subprocess.run(call_cmd, timeout=300, text=True, capture_output=True, check=False)
|
|
|
|
|
if process.stdout.strip() != "":
|
|
|
|
|
logger.warning(process.stdout.strip())
|
|
|
|
|
try:
|
|
|
|
|
process = subprocess.run(call_cmd, timeout=3, text=True, capture_output=True, check=False)
|
|
|
|
|
if process.stdout.strip() != "":
|
|
|
|
|
logger.warning(process.stdout.strip())
|
|
|
|
|
except subprocess.TimeoutExpired:
|
|
|
|
|
logger.warning("Package te, topi, hccl version check timed out, skip.")
|
|
|
|
|
|
|
|
|
|
def set_env(self):
|
|
|
|
|
if not self.tbe_path:
|
|
|
|
|