include sm70

opentensor · Oct 3, 2022 · ce9fd7f · ce9fd7f
1 parent 761a3e7
commit ce9fd7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -138,7 +138,7 @@ def build_extensions(self):
             extra_compile_args={
                 'gcc': [],
                 'nvcc': ['-Xptxas', '-v', '-O3',
-                '-gencode', 'arch=compute_86,code=sm_86', '-gencode', 'arch=compute_80,code=sm_80', '-gencode', 'arch=compute_75,code=sm_75', '--ptxas-options=-v', '-c',
+                '-gencode', 'arch=compute_86,code=sm_86', '-gencode', 'arch=compute_80,code=sm_80', '-gencode', 'arch=compute_75,code=sm_75', '-gencode', 'arch=compute_70,code=sm_70', '--ptxas-options=-v', '-c',
                 '--compiler-options', "'-fPIC'"
                 ]
             },