Update to CUDA 12.8.0. (#121)

Updates pynvjitlink to CUDA 12.8.0. --------- Co-authored-by: jakirkham <[email protected]>
rapidsai · Jan 27, 2025 · 6f9deb3 · 6f9deb3
1 parent 32094d5
commit 6f9deb3
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 7 deletions.
diff --git a/pynvjitlink/CUDA_VERSION b/pynvjitlink/CUDA_VERSION
@@ -1 +1 @@
-12.6
+12.8
diff --git a/pynvjitlink/tests/conftest.py b/pynvjitlink/tests/conftest.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 import os
 
@@ -28,7 +28,7 @@ def alt_gpu_compute_capability(gpu_compute_capability):
 @pytest.fixture(scope="session")
 def absent_gpu_compute_capability(gpu_compute_capability, alt_gpu_compute_capability):
     """A compute capability not used in any cubin or fatbin test binary"""
-    cc_majors = {6, 7, 8}
+    cc_majors = {6, 7, 8, 9, 10, 12}
     cc_majors.remove(gpu_compute_capability[0])
     cc_majors.remove(alt_gpu_compute_capability[0])
     return (cc_majors.pop(), 0)

diff --git a/pynvjitlink/tests/test_pynvjitlink.py b/pynvjitlink/tests/test_pynvjitlink.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2023-2024, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2023-2025, NVIDIA CORPORATION. All rights reserved.
 
 import pynvjitlink
 import pytest
@@ -84,7 +84,11 @@ def test_get_error_log(undefined_extern_cubin, gpu_arch_flag):
     )
 
 
-def test_get_info_log(device_functions_cubin, gpu_arch_flag):
+def test_get_info_log(device_functions_cubin, gpu_arch_flag, gpu_compute_capability):
+    if gpu_compute_capability < (7, 5):
+        pytest.skip(
+            "CUDA 12.8 shows deprecations for devices older than compute capability 7.5"
+        )
     handle = _nvjitlinklib.create(gpu_arch_flag)
     filename, data = device_functions_cubin
     input_type = InputType.CUBIN.value

diff --git a/pynvjitlink/tests/test_pynvjitlink_api.py b/pynvjitlink/tests/test_pynvjitlink_api.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2023-2024, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2023-2025, NVIDIA CORPORATION. All rights reserved.
 
 import sys
 
@@ -120,7 +120,11 @@ def test_get_error_log(undefined_extern_cubin, gpu_arch_flag):
     assert "Undefined reference to '_Z5undefff'" in error_log
 
 
-def test_get_info_log(device_functions_cubin, gpu_arch_flag):
+def test_get_info_log(device_functions_cubin, gpu_arch_flag, gpu_compute_capability):
+    if gpu_compute_capability < (7, 5):
+        pytest.skip(
+            "CUDA 12.8 shows deprecations for devices older than compute capability 7.5"
+        )
     nvjitlinker = NvJitLinker(gpu_arch_flag)
     name, cubin = device_functions_cubin
     nvjitlinker.add_cubin(cubin, name)