43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
|
#
|
||
|
|
# Author: Joshua Cohen
|
||
|
|
# Copyright 2016
|
||
|
|
#
|
||
|
|
# This setup file will compile the relevant C++ files against the provided gpu_topozero.pyx
|
||
|
|
# file to create a gpu_topozero module that can be imported in Python and used as an
|
||
|
|
# interface for the PyTopozero() object to run the Topo C++ code.
|
||
|
|
|
||
|
|
from distutils.core import setup
|
||
|
|
from distutils.extension import Extension # Normally not needed but we need to add the
|
||
|
|
# extra c++11, fopenmp, and lgomp flags
|
||
|
|
from Cython.Build import cythonize
|
||
|
|
|
||
|
|
# Where the .cpp files are located
|
||
|
|
source_dir = "src/"
|
||
|
|
# All files contained in source_dir
|
||
|
|
source_files = ["AkimaLib.cpp",
|
||
|
|
"Ellipsoid.cpp",
|
||
|
|
"LinAlg.cpp",
|
||
|
|
"Orbit.cpp",
|
||
|
|
"Peg.cpp",
|
||
|
|
"PegTrans.cpp",
|
||
|
|
"Poly2d.cpp",
|
||
|
|
#"Position.cpp", Leaving this out for now as it's not being used
|
||
|
|
"Topo.cpp",
|
||
|
|
"TopoMethods.cpp",
|
||
|
|
"Topozero.cpp",
|
||
|
|
"UniformInterp.cpp"]
|
||
|
|
source_files = [(source_dir + f) for f in source_files] # Quick one-line to prepend the source_dir
|
||
|
|
|
||
|
|
setup(ext_modules = cythonize(Extension(
|
||
|
|
"gpu_topozero", # Name of the module
|
||
|
|
sources=['gpu_topozero.pyx'] + source_files, # Source files (.cpp and .pyx)
|
||
|
|
include_dirs=['include/', # Header files (.h)
|
||
|
|
'../../iscesys/ImageApi/InterleavedAccessor/include/',
|
||
|
|
'../../iscesys/ImageApi/DataCaster/include/'],
|
||
|
|
extra_compile_args=['-fopenmp','-std=c++11','-fPIC','-pthread'], # Allows for OMP and special libraries
|
||
|
|
extra_objects=['gpu-topo.o'],
|
||
|
|
extra_link_args=['-lgomp','-lpthread','-L/usr/local/cuda/lib64','-lcudart'], # Needed to link the OMP/CUDA libraries in
|
||
|
|
language="c++"
|
||
|
|
)))
|
||
|
|
|