未验证 提交 585f9d65 编写于 作者: Q Qi Li 提交者: GitHub

Update default custom device dir for version check, test=develop (#53020) (#53031)

上级 e7848c31
......@@ -378,7 +378,7 @@ def set_paddle_lib_path():
if os.path.exists(lib_dir):
_set_paddle_lib_path(lib_dir)
set_paddle_custom_device_lib_path(
os.path.sep.join([lib_dir, '..', '..', 'paddle-plugins'])
os.path.sep.join([lib_dir, '..', '..', 'paddle_custom_device'])
)
return
if hasattr(site, 'USER_SITE'):
......@@ -386,7 +386,7 @@ def set_paddle_lib_path():
if os.path.exists(lib_dir):
_set_paddle_lib_path(lib_dir)
set_paddle_custom_device_lib_path(
os.path.sep.join([lib_dir, '..', '..', 'paddle-plugins'])
os.path.sep.join([lib_dir, '..', '..', 'paddle_custom_device'])
)
......
......@@ -51,7 +51,7 @@ class TestCustomKernelLoad(unittest.TestCase):
if os.path.exists(lib_dir):
paddle_lib_path = lib_dir
self.default_path = os.path.sep.join(
[paddle_lib_path, '..', '..', 'paddle-plugins']
[paddle_lib_path, '..', '..', 'paddle_custom_device']
)
# copy so to default path
cmd = 'mkdir -p {} && cp ./*.so {}'.format(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册