Commit 15441ada9881da2d14b57fa1471f642c311c012d
1 parent
a7b63368
Exists in
master
and in
3 other branches
Only using compiling with openmp on Linux
Showing
1 changed file
with
16 additions
and
7 deletions
Show diff stats
setup.py
@@ -2,12 +2,21 @@ from distutils.core import setup | @@ -2,12 +2,21 @@ from distutils.core import setup | ||
2 | from distutils.extension import Extension | 2 | from distutils.extension import Extension |
3 | from Cython.Distutils import build_ext | 3 | from Cython.Distutils import build_ext |
4 | 4 | ||
5 | +import sys | ||
6 | + | ||
5 | import numpy | 7 | import numpy |
6 | 8 | ||
7 | -setup( | ||
8 | - cmdclass = {'build_ext': build_ext}, | ||
9 | - ext_modules = [ Extension("invesalius/data/mips", ["invesalius/data/mips.pyx"], | ||
10 | - include_dirs = [numpy.get_include()], | ||
11 | - extra_compile_args=['-fopenmp'], | ||
12 | - extra_link_args=['-fopenmp'],)] | ||
13 | - ) | 9 | +if sys.platform == 'linux2': |
10 | + setup( | ||
11 | + cmdclass = {'build_ext': build_ext}, | ||
12 | + ext_modules = [ Extension("invesalius.data.mips", ["invesalius/data/mips.pyx"], | ||
13 | + include_dirs = [numpy.get_include()], | ||
14 | + extra_compile_args=['-fopenmp'], | ||
15 | + extra_link_args=['-fopenmp'],)] | ||
16 | + ) | ||
17 | +else: | ||
18 | + setup( | ||
19 | + cmdclass = {'build_ext': build_ext}, | ||
20 | + ext_modules = [ Extension("invesalius.data.mips", ["invesalius/data/mips.pyx"], | ||
21 | + include_dirs = [numpy.get_include()],)] | ||
22 | + ) |