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

numba_cuda/numba/cuda/tests/test_binary_generation/Makefile

@@ -1,14 +1,17 @@
-# Generates the input files used by the pynvjitlink binding test suite
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+# 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,14 @@
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
 
 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 +79,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/tests/test_binary_generation/nrt_extern.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include <nrt.cuh>
 
 extern "C" __device__ int device_allocate_deallocate(int* nb_retval){

numba_cuda/numba/cuda/tests/test_binary_generation/test_device_functions.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 #include <cuda_fp16.h>
 
 extern __device__ bool __heq(__half arg1, __half arg2);

numba_cuda/numba/cuda/tests/test_binary_generation/undefined_extern.cu

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
 extern __device__ float undef(float a, float b);
 
 __global__ void f(float *r, float *a, float *b) { r[0] = undef(a[0], b[0]); }

numba_cuda/numba/cuda/types.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
 from numba.core import types
 
 

numba_cuda/numba/cuda/typing/__init__.py

@@ -0,0 +1,11 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+from .templates import (
+    signature,
+    make_concrete_template,
+    Signature,
+    fold_arguments,
+)
+
+__all__ = ["signature", "make_concrete_template", "Signature", "fold_arguments"]