numba-cuda 0.18.1__py3-none-any.whl → 0.19.0__py3-none-any.whl

numba_cuda/numba/cuda/tests/test_binary_generation/Makefile

@@ -1,14 +1,14 @@
-# Generates the input files used by the pynvjitlink binding test suite
+# Generates the input files used by the nvjitlink tests
 
 # Test binaries are built taking into account the CC of the GPU in the test machine
 GPU_CC := $(shell nvidia-smi --query-gpu=compute_cap --format=csv | grep -v compute_cap | head -n 1 | sed 's/\.//')
 GPU_CC ?= 75
 
-# Use CC 7.0 as an alternative in fatbin testing, unless CC is 7.x
+# Use CC 7.5 as an alternative in fatbin testing, unless CC is 7.x
 ifeq ($(shell echo "$(GPU_CC)" | cut -c1),7)
     ALT_CC := 80
 else
-    ALT_CC := 70
+    ALT_CC := 75
 endif
 
 # Gencode flags suitable for most tests

numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py

@@ -1,12 +1,13 @@
 # Copyright (c) 2024, NVIDIA CORPORATION.
 
 import argparse
+import os
 import pathlib
 import platform
 import subprocess
 import sys
 
-from cuda import nvrtc
+from cuda.bindings import nvrtc
 from numba.cuda.memory_management.nrt import get_include
 
 # Magic number found at the start of an LTO-IR file
@@ -77,6 +78,8 @@ def determine_include_flags():
 
     quoted_flags = includes_lines[0].split("INCLUDES=")[1].strip().split()
     include_flags = [flag.strip('"') for flag in quoted_flags]
+    cccl_include_flags = [flag + os.path.sep + "cccl" for flag in include_flags]
+    include_flags += cccl_include_flags
     print(f"Using CUDA include flags: {include_flags}")
 
     return include_flags

numba_cuda/numba/cuda/typing/__init__.py

@@ -0,0 +1,8 @@
+from .templates import (
+    signature,
+    make_concrete_template,
+    Signature,
+    fold_arguments,
+)
+
+__all__ = ["signature", "make_concrete_template", "Signature", "fold_arguments"]