diff --git a/setup.py b/setup.py index b1b08b3..2ca06f9 100644 --- a/setup.py +++ b/setup.py @@ -42,6 +42,7 @@ cc_flag = [ "-gencode", "arch=compute_120a,code=sm_120a", + "-gencode", "arch=compute_100,code=sm_100", "-gencode", "arch=compute_90,code=sm_90", "-gencode", "arch=compute_89,code=sm_89", "-gencode", "arch=compute_80,code=sm_80" @@ -76,4 +77,4 @@ ), ext_modules=ext_modules, cmdclass={"build_ext": BuildExtension}, -) \ No newline at end of file +)