diff --git a/cuda_bindings_13/tests/test_nvfatbin.py b/cuda_bindings_13/tests/test_nvfatbin.py index 1bb8d4fcc53..ebb8fda91c9 100644 --- a/cuda_bindings_13/tests/test_nvfatbin.py +++ b/cuda_bindings_13/tests/test_nvfatbin.py @@ -12,7 +12,7 @@ from cuda.bindings import nvfatbin from cuda.bindings._v2 import nvrtc -ARCHITECTURES = ["sm_75", "sm_80", "sm_90", "sm_100"] +ARCHITECTURES = ["sm_75", "sm_80", "sm_90", "sm_90a", "sm_100", "sm_103", "sm_103a"] PTX_VERSIONS = ["6.4", "7.0", "8.5", "8.8"] PTX_TEMPLATE = """ diff --git a/cuda_bindings_13/tests/test_nvjitlink.py b/cuda_bindings_13/tests/test_nvjitlink.py index 3fa6837a9a0..196263980c5 100644 --- a/cuda_bindings_13/tests/test_nvjitlink.py +++ b/cuda_bindings_13/tests/test_nvjitlink.py @@ -21,8 +21,8 @@ def nvjitlink_session(num_options, options): # Establish a handful of compatible architectures and PTX versions to test with -ARCHITECTURES = ["sm_75", "sm_80", "sm_90", "sm_100"] -PTX_VERSIONS = ["6.4", "7.0", "8.5", "8.8"] +ARCHITECTURES = ["sm_75", "sm_80", "sm_90", "sm_90a", "sm_100", "sm_103", "sm_103a"] +PTX_VERSIONS = ["6.4", "7.0", "8.5", "8.0", "8.8", "8.8", "8.8"] PTX_HEADER = """\