From ce9fd7f371914c1012d02c7b30b56e2abd536d97 Mon Sep 17 00:00:00 2001 From: Cameron Fairchild Date: Mon, 3 Oct 2022 18:50:21 -0400 Subject: [PATCH] include sm70 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 53e5a63..e860ced 100644 --- 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'" ] },