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

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

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