Skip to content

Commit 585f9d6

Browse files
authored
Update default custom device dir for version check, test=develop (#53020) (#53031)
1 parent e7848c3 commit 585f9d6

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/paddle/fluid/core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -378,15 +378,15 @@ def set_paddle_lib_path():
378378
if os.path.exists(lib_dir):
379379
_set_paddle_lib_path(lib_dir)
380380
set_paddle_custom_device_lib_path(
381-
os.path.sep.join([lib_dir, '..', '..', 'paddle-plugins'])
381+
os.path.sep.join([lib_dir, '..', '..', 'paddle_custom_device'])
382382
)
383383
return
384384
if hasattr(site, 'USER_SITE'):
385385
lib_dir = os.path.sep.join([site.USER_SITE, 'paddle', 'libs'])
386386
if os.path.exists(lib_dir):
387387
_set_paddle_lib_path(lib_dir)
388388
set_paddle_custom_device_lib_path(
389-
os.path.sep.join([lib_dir, '..', '..', 'paddle-plugins'])
389+
os.path.sep.join([lib_dir, '..', '..', 'paddle_custom_device'])
390390
)
391391

392392

test/custom_kernel/test_custom_kernel_load.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def setUp(self):
5151
if os.path.exists(lib_dir):
5252
paddle_lib_path = lib_dir
5353
self.default_path = os.path.sep.join(
54-
[paddle_lib_path, '..', '..', 'paddle-plugins']
54+
[paddle_lib_path, '..', '..', 'paddle_custom_device']
5555
)
5656
# copy so to default path
5757
cmd = 'mkdir -p {} && cp ./*.so {}'.format(

0 commit comments

Comments
 (0)