-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
41 lines (27 loc) · 935 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from setuptools import Command, Extension, setup
def define_extensions():
compile_args = ['-ffast-math', '-march=native', '-std=c11']
return [Extension("binge.libpredict",
['binge/predict.c'],
extra_compile_args=compile_args)]
class BuildExtension(Command):
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
from netrex.native import _build_module
_build_module()
setup(
name='binge',
version='0.1.0',
requirements=['pytorch==0.1.11'],
packages=['binge'],
license='MIT',
cmdclass={'build_extension': BuildExtension},
classifiers=['Development Status :: 3 - Alpha',
'License :: OSI Approved :: MIT License',
'Topic :: Scientific/Engineering :: Artificial Intelligence'],
ext_modules=define_extensions()
)