feat: add new archs in new cuda releases (#325)

bazel-contrib · Feb 20, 2025 · 3241422 · 3241422
1 parent 6a6d2f1
commit 3241422
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/cuda/private/providers.bzl b/cuda/private/providers.bzl
@@ -20,6 +20,12 @@ cuda_archs = [
     "89",
     "90",
     "90a",
+    "100",
+    "100a",
+    "101",
+    "101a",
+    "120",
+    "120a",
 ]
 
 Stage2ArchInfo = provider(

diff --git a/tests/flag/BUILD.bazel b/tests/flag/BUILD.bazel
@@ -11,6 +11,7 @@ load(
     "cuda_library_compute60_sm61_flag_test",
     "cuda_library_compute61_sm61_flag_test",
     "cuda_library_flag_test",
+    "cuda_library_sm100_sm100a_flag_test",
     "cuda_library_sm61_flag_test",
     "cuda_library_sm90a_flag_test",
     "cuda_library_sm90a_sm90_flag_test",
@@ -348,3 +349,13 @@ cuda_library_sm90a_sm90_flag_test(
     ],
     target_under_test = "@rules_cuda_examples//basic:kernel",
 )
+
+cuda_library_sm100_sm100a_flag_test(
+    name = "cuda_library_sm100_sm100a_arch_flag_test",
+    action_mnemonic = "CudaCompile",
+    contain_flags = [
+        "-gencode arch=compute_100,code=sm_100",
+        "-gencode arch=compute_100a,code=sm_100a",
+    ],
+    target_under_test = "@rules_cuda_examples//basic:kernel",
+)
diff --git a/tests/flag/flag_validation_test.bzl b/tests/flag/flag_validation_test.bzl
@@ -104,10 +104,12 @@ config_settings_compute60_sm61 = {_rules_cuda_target("cuda:archs"): "compute_60,
 config_settings_compute61_sm61 = {_rules_cuda_target("cuda:archs"): "compute_61,sm_61"}
 config_settings_sm90a = {_rules_cuda_target("cuda:archs"): "sm_90a"}
 config_settings_sm90a_sm90 = {_rules_cuda_target("cuda:archs"): "sm_90a,sm_90"}
+config_settings_sm100_sm100a = {_rules_cuda_target("cuda:archs"): "sm_100;sm_100a"}  # NOTE: two specs
 
 cuda_library_sm61_flag_test = _create_cuda_library_flag_test(config_settings_sm61)
 cuda_library_sm90a_flag_test = _create_cuda_library_flag_test(config_settings_sm90a)
 cuda_library_sm90a_sm90_flag_test = _create_cuda_library_flag_test(config_settings_sm90a_sm90)
+cuda_library_sm100_sm100a_flag_test = _create_cuda_library_flag_test(config_settings_sm100_sm100a)
 cuda_library_compute60_flag_test = _create_cuda_library_flag_test(config_settings_compute60)
 cuda_library_compute60_sm61_flag_test = _create_cuda_library_flag_test(config_settings_compute60_sm61)
 cuda_library_compute61_sm61_flag_test = _create_cuda_library_flag_test(config_settings_compute61_sm61)
-Original file line number
+Diff line change
@@ Expand Up / @@ -20,6 +20,12 @@ cuda_archs = [ @@
         "89",
         "90",
         "90a",
+        "100",
+        "100a",
+        "101",
+        "101a",
+        "120",
+        "120a",
     ]
     Stage2ArchInfo = provider(
@@ Expand Down @@