diff --git a/python/paddle/fluid/core.py b/python/paddle/fluid/core.py index 508532e51e83ac26b26017394bd8deca0427d849..a07e12f199c7e3b7600dcb63a4d432225b66daf0 100644 --- a/python/paddle/fluid/core.py +++ b/python/paddle/fluid/core.py @@ -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']) ) diff --git a/test/custom_kernel/test_custom_kernel_load.py b/test/custom_kernel/test_custom_kernel_load.py index 9428a6d0d454a79d3fe9a400f8d32b7a0789a19d..f790ae9e3ed9df00582c798216a37221b73a31cb 100644 --- a/test/custom_kernel/test_custom_kernel_load.py +++ b/test/custom_kernel/test_custom_kernel_load.py @@ -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(