custom op env vars setting

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
pull/11171/head
zhoufeng 4 years ago
parent d252fc102c
commit 12d32a5375

@ -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(

Loading…
Cancel
Save