Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 10 additions & 6 deletions numba_cuda/numba/cuda/tests/cudapy/test_ir_utils.py
Original file line number Diff line number Diff line change
@@ -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,
Expand All @@ -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.
Expand All @@ -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,
Expand Down