add psenet

2026-09-24 23:33:08 +08:00 · 2021-07-27 15:39:12 +08:00
parent e132f67005
commit 017ff26feb
1 changed files with 14 additions and 0 deletions
@@ -0,0 +1,14 @@
+from distutils.core import setup, Extension
+from Cython.Build import cythonize
+import numpy
+
+setup(ext_modules=cythonize(Extension(
+    'pse',
+    sources=['pse.pyx'],
+    language='c++',
+    include_dirs=[numpy.get_include()],
+    library_dirs=[],
+    libraries=[],
+    extra_compile_args=['-O3'],
+    extra_link_args=[]
+)))