1
0
mirror of https://github.com/exaloop/codon.git synced 2025-06-03 15:03:52 +08:00
codon/extra/python/setup.py
A. R. Shajii f4323a80a2
Fix miscellaneous issues (#85)
* Fix #75

* Build on macOS 11 for #72

* Fix #76

* Fix test

* Fix generator example in docs

* Fix Python/Cython packaging; Fix #80

* Fix tests

* Fix tests

* Fix tests

* Fix tests

* Fix syntax

* Fix CI

* Fix CI

* Fix CI; Dominate imported bindings

* Fix CI; Dominate imported bindings

* Fix .gitignore

* Fix tests

* Fix rm command

* Format

* clang-format

* Organize and optimize string formatting

* Fix format error message

Co-authored-by: Ibrahim Numanagić <ibrahimpasa@gmail.com>
2022-12-12 20:54:01 -05:00

81 lines
2.1 KiB
Python

# Copyright (C) 2022 Exaloop Inc. <https://exaloop.io>
import os
import sys
import shutil
from pathlib import Path
from Cython.Distutils import build_ext
from setuptools import setup
from setuptools.extension import Extension
exec(open("codon/version.py").read())
ext = "dylib" if sys.platform == "darwin" else "so"
codon_path = os.environ.get("CODON_DIR")
if not codon_path:
c = shutil.which("codon")
if c:
codon_path = Path(c).parent / ".."
else:
codon_path = Path(codon_path)
for path in [
os.path.expanduser("~") + "/.codon",
os.getcwd() + "/..",
]:
path = Path(path)
if not codon_path and path.exists():
codon_path = path
break
if (
not codon_path
or not (codon_path / "include" / "codon").exists()
or not (codon_path / "lib" / "codon").exists()
):
print(
"Cannot find Codon.",
'Please either install Codon (/bin/bash -c "$(curl -fsSL https://exaloop.io/install.sh)"),',
"or set CODON_DIR if Codon is not in PATH or installed in ~/.codon",
file=sys.stderr,
)
sys.exit(1)
codon_path = codon_path.resolve()
print("Codon: " + str(codon_path))
if sys.platform == "darwin":
linker_args = "-Wl,-rpath," + str(codon_path / "lib" / "codon")
else:
linker_args = "-Wl,-rpath=" + str(codon_path / "lib" / "codon")
jit_extension = Extension(
"codon.codon_jit",
sources=["codon/jit.pyx", "codon/jit.pxd"],
libraries=["codonc", "codonrt"],
language="c++",
extra_compile_args=["-w", "-std=c++17"],
extra_link_args=[linker_args],
library_dirs=[str(codon_path / "lib" / "codon")],
)
setup(
name="codon",
version=__version__,
install_requires=["cython", "astunparse"],
python_requires=">=3.6",
description="Codon JIT decorator",
url="https://exaloop.io",
long_description="Please see https://exaloop.io for more details.",
author="Exaloop Inc.",
author_email="info@exaloop.io",
license="Commercial",
ext_modules=[jit_extension],
packages=["codon"],
include_package_data=True,
cmdclass={
"build_ext": build_ext,
},
)