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.
|
|
|
|
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
|
|
|
|
|
)
|
|
|
|
|
]
|
|
|
|
|
)
|
