From 585f9d652d192b05f0765b208b5f031f528676f3 Mon Sep 17 00:00:00 2001 From: Qi Li Date: Thu, 20 Apr 2023 11:51:08 +0800 Subject: [PATCH] Update default custom device dir for version check, test=develop (#53020) (#53031) --- python/paddle/fluid/core.py | 4 ++-- test/custom_kernel/test_custom_kernel_load.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/core.py b/python/paddle/fluid/core.py index 508532e51e8..a07e12f199c 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 9428a6d0d45..f790ae9e3ed 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( -- GitLab