-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
41 lines (32 loc) · 1.02 KB
/
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
#!/usr/bin/env python
from pybind11.setup_helpers import Pybind11Extension, build_ext, intree_extensions
from setuptools import setup
__version__ = "0.0.1"
setup_helper = Pybind11Extension(
"eml",
["src/pyeml/lib/eml.cpp"],
)
#, "-Wl,--as-needed", "-lnvidia-ml" , "-Wl,--no-as-needed"
setup_helper._add_cflags(["-shared"])
setup_helper._add_ldflags(["-L/usr/local/lib", "-leml" ,"-lconfuse", "-std=c++11"])
setup(
name="pyeml",
version=__version__,
author="Daniel Suárez Labena",
author_email="[email protected]",
url="https://github.com/dsuarezl/Pyeml",
description="Tool to measure energy consumption of python code",
long_description="",
ext_modules=[setup_helper],
packages = ["pyeml",
"pyeml.devices",
"pyeml.units"
],
package_dir={'pyeml': 'src/pyeml',
'pyeml.devices': 'src/pyeml/devices',
'pyeml.units': 'src/pyeml/units',
},
cmdclass={"build_ext": build_ext},
zip_safe=False,
python_requires=">=3.7",
)