Add support for H100/sm_90 arch compilation

microsoft · Oct 25, 2024 · 5a76cfa · 5a76cfa
1 parent 3d5cf73
commit 5a76cfa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -67,7 +67,7 @@ def get_default_compute_capabilities():
             # Special treatment of CUDA 11.0 because compute_86 is not supported.
             compute_caps += ";8.0"
         else:
-            compute_caps += ";8.0;8.6"
+            compute_caps += ";8.0;8.6;9.0"
     return compute_caps