add sm_89 (#45)

microsoft · Oct 6, 2024 · c59a20b · c59a20b
1 parent 0520d7c
commit c59a20b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@ def get_version():
 
 
 def build_cuda_extensions():
-    compute_capabilities = [70, 75, 80, 86, 90]
+    compute_capabilities = [70, 75, 80, 86, 89, 90]
     arch_flags = []
     TORCH_CUDA_ARCH_LIST = os.getenv("TORCH_CUDA_ARCH_LIST", None)
     if TORCH_CUDA_ARCH_LIST is None: