euler314 commited on
Commit
6c21feb
·
verified ·
1 Parent(s): 9a06845

Create setup.py

Browse files
Files changed (1) hide show
  1. setup.py +23 -0
setup.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from setuptools import setup, Extension
2
+ import pybind11
3
+ import numpy as np
4
+
5
+ ext_modules = [
6
+ Extension(
7
+ 'cubic_cpp',
8
+ ['cubic_cpp.cpp'],
9
+ include_dirs=[pybind11.get_include(), np.get_include()],
10
+ language='c++',
11
+ extra_compile_args=['-std=c++11', '-O3'],
12
+ ),
13
+ ]
14
+
15
+ setup(
16
+ name="cubic_cpp",
17
+ version="0.1",
18
+ ext_modules=ext_modules,
19
+ install_requires=[
20
+ 'pybind11>=2.6.0',
21
+ 'numpy>=1.19.0',
22
+ ],
23
+ )