Skip to content

Commit 15dc22a

Browse files
committed
add an extra guard in test_overload for cpu tests
1 parent 31af244 commit 15dc22a

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

numba_cuda/numba/cuda/tests/cudapy/test_overload.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# SPDX-License-Identifier: BSD-2-Clause
33

44
from numba import cuda
5-
import numba.cuda.types as types
5+
from numba.cuda import types
66
from numba.cuda import _HAS_NUMBA
77

88
if _HAS_NUMBA:
@@ -338,7 +338,12 @@ def kernel(x):
338338
expected = CUDA_TARGET_OL_CALLS_TARGET_OL * CUDA_TARGET_OL
339339
self.check_overload(kernel, expected)
340340

341-
# Also check that the CPU overloads are used on the CPU
341+
@skip_on_standalone_numba_cuda
342+
def test_target_overloaded_calls_target_overloaded_cpu(self):
343+
def kernel(x):
344+
target_overloaded_calls_target_overloaded(x)
345+
346+
# Check that the CPU overloads are used on the CPU
342347
expected = GENERIC_TARGET_OL_CALLS_TARGET_OL * GENERIC_TARGET_OL
343348
self.check_overload_cpu(kernel, expected)
344349

0 commit comments

Comments
 (0)