diff --git a/setup.py b/setup.py index e2723268..fe130d50 100644 --- a/setup.py +++ b/setup.py @@ -23,6 +23,7 @@ 'cxx': ['-O3'], 'nvcc': [ '-O3', + '-gencode=arch=compute_86,code=sm_86', '-gencode=arch=compute_89,code=sm_89', '-gencode=arch=compute_90,code=sm_90' ],