diff --git a/numba_cuda/numba/cuda/tests/cudapy/test_ir_utils.py b/numba_cuda/numba/cuda/tests/cudapy/test_ir_utils.py index 4ff63b186..b03e9c2e3 100644 --- a/numba_cuda/numba/cuda/tests/cudapy/test_ir_utils.py +++ b/numba_cuda/numba/cuda/tests/cudapy/test_ir_utils.py @@ -1,12 +1,12 @@ # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: BSD-2-Clause -from numba.cuda.testing import CUDATestCase +from numba.cuda.testing import CUDATestCase, skip_on_cudasim from numba.cuda.core.compiler import CompilerBase from numba.cuda.flags import Flags from numba.cuda.core.compiler_machinery import PassManager from numba.cuda.core import ir_utils -from numba.core import types, ir, bytecode, registry +from numba.core import types, ir, bytecode from numba.cuda import compiler from numba.cuda.core.untyped_passes import ( ExtractByteCode, @@ -18,13 +18,13 @@ NopythonTypeInference, DeadCodeElimination, ) -from numba.cuda.testing import skip_on_cudasim # global constant for testing find_const GLOBAL_B = 11 +@skip_on_cudasim("Requires CUDA target") class TestIrUtils(CUDATestCase): """ Tests ir handling utility functions like find_callname. @@ -47,11 +47,15 @@ def mk_pipeline( locals = {} if not flags: flags = Flags() - flags.nrt = True + if typing_context is None: - typing_context = registry.cpu_target.typing_context + from numba.cuda.descriptor import cuda_target + + typing_context = cuda_target.typing_context if target_context is None: - target_context = registry.cpu_target.target_context + from numba.cuda.descriptor import cuda_target + + target_context = cuda_target.target_context return cls( typing_context, target_context,