forked from halide/Halide
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
34 lines (32 loc) · 1.24 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
import pybind11
from setuptools import find_packages
from skbuild import setup
from pathlib import Path
this_directory = Path(__file__).parent
long_description = (this_directory / "README_python.md").read_text()
setup(
name="halide",
version='17.0.0',
author="The Halide team",
author_email="[email protected]",
description="Halide is a programming language designed to make it easier "
"to write high-performance image and array processing code.",
long_description=long_description,
long_description_content_type='text/markdown',
python_requires=">=3.8",
packages=find_packages(where="python_bindings/src"),
package_dir={"": "python_bindings/src"},
cmake_source_dir="python_bindings",
cmake_args=[
f"-Dpybind11_ROOT={pybind11.get_cmake_dir()}",
"-DCMAKE_REQUIRE_FIND_PACKAGE_pybind11=YES",
"-DHalide_INSTALL_PYTHONDIR=python_bindings/src",
"-DCMAKE_INSTALL_RPATH=$<IF:$<PLATFORM_ID:Darwin>,@loader_path,$ORIGIN>",
"-DHalide_Python_INSTALL_IMPORTED_DEPS=ON",
"-DWITH_TESTS=NO",
"-DWITH_TUTORIALS=NO",
"-DWITH_PYTHON_STUBS=NO",
"-DCMAKE_PREFIX_PATH=$ENV{CMAKE_PREFIX_PATH}",
"--no-warn-unused-cli",
],
)