diff --git a/setup.py b/setup.py
index 275b6fc3..3e14ad59 100644
--- a/setup.py
+++ b/setup.py
@@ -92,6 +92,7 @@ def get_extensions():
             "-D__CUDA_NO_HALF_OPERATORS__",
             "-D__CUDA_NO_HALF_CONVERSIONS__",
             "-D__CUDA_NO_HALF2_OPERATORS__",
+            "-gencode=arch=compute_120,code=sm_120",
         ]
     else:
         print("Compiling without CUDA")