Skip to content

Commit e45f093

Browse files
author
Jeff Daily
committed
generalize build for parallel cython impl
1 parent 20d880b commit e45f093

2 files changed

Lines changed: 38 additions & 4 deletions

File tree

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
python setup.py build_ext --inplace

‎source code/Parallel Cython/setup.py‎

Lines changed: 37 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,41 @@
11
from distutils.core import setup
2-
from Cython.Build import cythonize
2+
from distutils.extension import Extension
33

4+
# numpy is required -- attempt import
5+
try:
6+
import numpy
7+
except ImportError:
8+
print "numpy is required"
9+
raise
10+
11+
# mpi4py is required -- attempt import
12+
try:
13+
import mpi4py
14+
except ImportError:
15+
print "mpi4py is required"
16+
raise
17+
18+
# cython is required -- attempt import
19+
try:
20+
from Cython.Build import cythonize
21+
from Cython.Distutils import build_ext
22+
except:
23+
print "cython is required"
24+
raise
25+
26+
include_dirs = [numpy.get_include(), '.']
27+
28+
ext_modules = [
29+
Extension(
30+
name="jac_2d",
31+
sources=["jac_2d.pyx"],
32+
include_dirs=include_dirs,
33+
),
34+
]
35+
36+
ext_modules = cythonize(ext_modules, include_path=include_dirs)
437
setup(
5-
name = 'MyProject',
6-
ext_modules = cythonize(["*.pyx"]),
38+
name="jac_2d",
39+
packages=[],
40+
ext_modules=ext_modules,
741
)
8-

0 commit comments

Comments
 (0)