From 4beef3bda490488a37c44189ab4bc82050ef2078 Mon Sep 17 00:00:00 2001 From: Stephan Seitz <stephan.seitz@fau.de> Date: Thu, 12 Mar 2020 15:19:32 +0100 Subject: [PATCH] Build CUDAExtension with '-03'. Maybe this helps? --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index e615a99..e774a6c 100644 --- a/setup.py +++ b/setup.py @@ -27,7 +27,7 @@ if __name__ == "__main__": CUDAExtension(module_name, [generated_file] + cuda_sources, extra_compile_args={'cxx': [], - 'nvcc': ['-arch=sm_35']}) + 'nvcc': ['-arch=sm_35', '-O3']}) ], cmdclass={ 'build_ext': BuildExtension -- GitLab