Merge pull request #292 from unnonouno/link-args

Fix link args for cusolver
cupy · Jul 21, 2017 · 5062f61 · 5062f61
2 parents aafd0eb + dcbdd41
commit 5062f61
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/cupy_setup_build.py b/cupy_setup_build.py
@@ -270,13 +270,15 @@ def make_extensions(options, compiler, use_cython):
             s['libraries'] = module['libraries']
 
         if module['name'] == 'cusolver':
-            args = s.setdefault('extra_compile_args', [])
+            compile_args = s.setdefault('extra_compile_args', [])
+            link_args = s.setdefault('extra_link_args', [])
             # openmp is required for cusolver
             if compiler.compiler_type == 'unix' and sys.platform != 'darwin':
                 # In mac environment, openmp is not required.
-                args.append('-fopenmp')
+                compile_args.append('-fopenmp')
+                link_args.append('-fopenmp')
             elif compiler.compiler_type == 'msvc':
-                args.append('/openmp')
+                compile_args.append('/openmp')
 
         if not no_cuda and module['name'] == 'thrust':
             if build.get_nvcc_path() is None: