Fix tests for CUDA 13.0

Author: mdboom

cuda_core/cuda/core/experimental/system/utils.pyx

@@ -20,7 +20,7 @@ cpdef str format_bytes(uint64_t x):
         return f"{x / 1024**4:.2f} TiB"
 
 
-cpdef list[int] unpack_bitmask(x: list[int]):
+cpdef list[int] unpack_bitmask(x: list[int] | array.array):
     """
     Unpack a list of integers containing bitmasks.
 

cuda_core/tests/system/test_system_device.py

@@ -10,6 +10,7 @@
 
 import os
 import re
+import sys
 
 import pytest
 from cuda.bindings import _nvml as nvml
@@ -30,7 +31,10 @@ def test_device_architecture():
         device_arch = device.architecture
 
         assert isinstance(device_arch, system_device.DeviceArchitecture)
-        assert device_arch.id in nvml.DeviceArch
+        if sys.version_info < (3, 12):
+            assert device_arch.id in nvml.DeviceArch.__members__.values()
+        else:
+            assert device_arch.id in nvml.DeviceArch
 
 
 def test_device_bar1_memory():

cuda_core/tests/system/test_system_system.py

@@ -58,7 +58,7 @@ def test_get_process_name():
         pytest.skip("Process not found")
 
     assert isinstance(process_name, str)
-    assert b"python" in process_name
+    assert "python" in process_name
 
 
 def test_device_count():