use setup CUDA flag, expt extended lambda (close #207)

NVIDIA · Dec 15, 2020 · 2f4ce0c · 2f4ce0c
1 parent e0799ce
commit 2f4ce0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -160,7 +160,7 @@ def _argparse(pattern, argv, is_flag=True):
 if "darwin" in platform:
     CC_FLAGS += ["-stdlib=libc++"]
 
-NVCC_FLAGS += ["--extended-lambda"]
+NVCC_FLAGS += ["--expt-extended-lambda"]
 
 BLAS_LIST = ["openblas", "mkl", "atlas", "blas"]
 if not (BLAS is False):  # False only when not set, str otherwise