Add support for H100/sm_90 arch compilation (#6669)

Resolves: #6549
microsoft · Oct 28, 2024 · 229960a · 229960a
1 parent 54903e0
commit 229960a
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