You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

49 lines
1.3 KiB

from distutils.core import setup
from Cython.Build import cythonize
from distutils.extension import Extension
from Cython.Distutils import build_ext
import numpy as np
import platform
import os
import json
this_dir = os.path.dirname(__file__)
with open('../config.json') as fp:
config = json.load(fp)
extra_compile_args = ['-O3', '-std=c++11']
print('using cuda')
cuda_lib_dir = config['CUDA_LIBRARY_DIR']
cuda_lib = 'cudart'
sources = ['cyrender.pyx']
extra_objects = [
os.path.join(this_dir, 'render/render_cpu.cpp.o'),
]
library_dirs = []
libraries = ['m']
extra_objects.append(os.path.join(this_dir, 'render/render_gpu.cu.o'))
extra_objects.append(os.path.join(this_dir, 'render/stdlib_cuda.cu.o'))
library_dirs.append(cuda_lib_dir)
libraries.append(cuda_lib)
setup(
name="cyrender",
cmdclass={'build_ext': build_ext},
ext_modules=[
Extension('cyrender',
sources,
extra_objects=extra_objects,
language='c++',
library_dirs=library_dirs,
libraries=libraries,
include_dirs=[
np.get_include(),
],
extra_compile_args=extra_compile_args,
# extra_link_args=extra_link_args
)
]
)