From 24019fb5ae6acda2188eb257a6e84bae79e9f0ff Mon Sep 17 00:00:00 2001 From: zhoufeng Date: Tue, 12 Jan 2021 09:52:44 +0800 Subject: [PATCH] custom op env vars setting Signed-off-by: zhoufeng --- mindspore/_check_version.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index bfa8d9b29d..8719d6800f 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -232,7 +232,14 @@ class AscendEnvChecker(EnvChecker): self.check_deps_version() if Path(self.op_impl_path).is_dir(): + # python path for sub process + if os.getenv('PYTHONPATH'): + os.environ['PYTHONPATH'] = self.op_impl_path + ":" + os.environ['PYTHONPATH'] + else: + os.environ['PYTHONPATH'] = self.op_impl_path + # sys path for this process sys.path.append(self.op_impl_path) + os.environ['TBE_IMPL_PATH'] = self.op_impl_path else: raise EnvironmentError(