mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 20:25:34 -05:00
47ef595a2a
Compiler tools such as CompilerEngine would now be imported from concrete.compiler Dialects would now be imported from concrete.lang.dialects And anything related to concretelang can be imported directly from concrete.lang
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
import os
|
|
import subprocess
|
|
import setuptools
|
|
import re
|
|
|
|
from setuptools import Extension
|
|
from setuptools.command.build_ext import build_ext
|
|
|
|
|
|
def read(fname):
|
|
return open(os.path.join(os.path.dirname(__file__), fname)).read()
|
|
|
|
|
|
def find_version():
|
|
return re.match(
|
|
r"__version__ = \"(?P<version>.+)\"",
|
|
read("lib/Bindings/Python/concrete/version.py"),
|
|
).group("version")
|
|
|
|
|
|
class MakeExtension(Extension):
|
|
def __init__(self, name, sourcedir=""):
|
|
Extension.__init__(self, name, sources=[])
|
|
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):
|
|
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(
|
|
name="concretefhe-compiler",
|
|
version=find_version(),
|
|
author="Zama Team",
|
|
author_email="hello@zama.ai",
|
|
description="Concrete Compiler",
|
|
license="",
|
|
keywords="homomorphic encryption compiler",
|
|
long_description=read("README.md"),
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/zama-ai/homomorphizer",
|
|
packages=setuptools.find_namespace_packages(
|
|
where=build_dir() + "tools/concretelang/python_packages/concretelang_core",
|
|
include=["concrete", "concrete.*"],
|
|
)
|
|
+ setuptools.find_namespace_packages(
|
|
where=build_dir() + "tools/concretelang/python_packages/concretelang_core",
|
|
include=["mlir", "mlir.*"],
|
|
),
|
|
install_requires=["numpy", "PyYAML"],
|
|
package_dir={"": build_dir() + "tools/concretelang/python_packages/concretelang_core"},
|
|
include_package_data=True,
|
|
package_data={"": ["*.so", "*.dylib"]},
|
|
classifiers=[
|
|
"Programming Language :: C++",
|
|
"Programming Language :: Python :: 3",
|
|
"Topic :: Software Development :: Compilers",
|
|
"Topic :: Security :: Cryptography",
|
|
],
|
|
ext_modules=[MakeExtension("python-bindings")],
|
|
cmdclass=dict(build_ext=MakeBuild),
|
|
zip_safe=False,
|
|
)
|