github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

refactor(python): customize python package build

Browse Source
This commit is contained in:
youben11
2021-12-08 20:55:25 +01:00
committed by Ayoub Benaissa
parent 16d0502f56
commit da8e6c3c0d
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
compiler/setup.py
View File

@@ -16,13 +16,25 @@ class MakeExtension(Extension):
self.sourcedir = os.path.abspath(sourcedir)
def build_dir():
return os.environ.get("CONCRETE_COMPILER_BUILD_DIR", "build/")
class MakeBuild(build_ext):
def run(self):
for ext in self.extensions:
self.build_extension(ext)
def build_extension(self, ext):
subprocess.check_call(["make", "python-bindings"])
cmd = ["make", f"BUILD_DIR={build_dir()}"]
py_exec = os.environ.get("CONCRETE_COMPILER_Python3_EXECUTABLE")
if py_exec:
cmd.append(f"Python3_EXECUTABLE={py_exec}")
ccache = os.environ.get("CONCRETE_COMPILER_CCACHE")
if ccache:
cmd.append(f"CCACHE={ccache}")
cmd.append("python-bindings")
subprocess.check_call(cmd)
setuptools.setup(
@@ -37,15 +49,15 @@ setuptools.setup(
long_description_content_type="text/markdown",
url="https://github.com/zama-ai/homomorphizer",
packages=setuptools.find_packages(
where="build/tools/zamalang/python_packages/zamalang_core",
where=build_dir() + "tools/zamalang/python_packages/zamalang_core",
include=["zamalang", "zamalang.*"],
)
+ setuptools.find_namespace_packages(
where="build/tools/zamalang/python_packages/zamalang_core",
where=build_dir() + "tools/zamalang/python_packages/zamalang_core",
include=["mlir", "mlir.*"],
),
install_requires=["numpy"],
package_dir={"": "build/tools/zamalang/python_packages/zamalang_core"},
package_dir={"": build_dir() + "tools/zamalang/python_packages/zamalang_core"},
include_package_data=True,
package_data={"": ["*.so"]},
classifiers=[