Skip to content

Commit 37a3d82

Browse files
authored
fix get_places (#74806)
1 parent 5fc05e3 commit 37a3d82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/legacy_test/op_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -410,8 +410,8 @@ def get_devices():
410410
if (
411411
os.environ.get('FLAGS_CI_both_cpu_and_gpu', 'False').lower()
412412
in ['1', 'true', 'on']
413-
or not paddle.is_compiled_with_cuda()
414-
):
413+
or not core.is_compiled_with_cuda()
414+
) and not is_custom_device():
415415
devices.append('cpu')
416416
if paddle.is_compiled_with_cuda():
417417
devices.append('gpu')

0 commit comments

Comments
 (0)