From 397f178a64a76222075528c105340d5b008d6c19 Mon Sep 17 00:00:00 2001 From: geyuqiang Date: Mon, 25 Aug 2025 06:52:48 +0000 Subject: [PATCH] fix get_places on custom device --- test/legacy_test/op_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/legacy_test/op_test.py b/test/legacy_test/op_test.py index a3ee5daa47551e..d7c7ff1408d2cc 100644 --- a/test/legacy_test/op_test.py +++ b/test/legacy_test/op_test.py @@ -402,7 +402,7 @@ def get_places(): os.environ.get('FLAGS_CI_both_cpu_and_gpu', 'False').lower() in ['1', 'true', 'on'] or not core.is_compiled_with_cuda() - ): + ) and not is_custom_device(): places.append(base.CPUPlace()) if core.is_compiled_with_cuda(): places.append(base.CUDAPlace(0))