mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-14 16:28:01 -05:00
fe618811ee
* Move most xfails to entries in tank/all_models.csv * enable usage of pytest without specifying tank/test_models.py * add dict_configs.py to gitignore. * Pin versions for runtimes and torch-mlir for setup.
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from setuptools import find_packages
|
|
from setuptools import setup
|
|
|
|
import os
|
|
|
|
with open("README.md", "r", encoding="utf-8") as fh:
|
|
long_description = fh.read()
|
|
|
|
with open("build_tools/shark_versions.txt", "r") as sv:
|
|
lines = [line.rstrip() for line in sv]
|
|
TM_VERSION = lines[7]
|
|
IREE_VERSION = lines[3]
|
|
|
|
|
|
PACKAGE_VERSION = os.environ.get("SHARK_PACKAGE_VERSION") or "0.0.4"
|
|
backend_deps = []
|
|
if "NO_BACKEND" in os.environ.keys():
|
|
backend_deps = [
|
|
f"iree-compiler=={IREE_VERSION}",
|
|
f"iree-runtime>={IREE_VERSION}",
|
|
]
|
|
|
|
setup(
|
|
name="nodai-SHARK",
|
|
version=f"{PACKAGE_VERSION}",
|
|
description="SHARK provides a High Performance Machine Learning Framework",
|
|
author="nod.ai",
|
|
author_email="stdin@nod.ai",
|
|
url="https://nod.ai",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
project_urls={
|
|
"Code": "https://github.com/nod-ai/SHARK",
|
|
"Bug Tracker": "https://github.com/nod-ai/SHARK/issues",
|
|
},
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
],
|
|
packages=find_packages(exclude=("examples")),
|
|
python_requires=">=3.9",
|
|
install_requires=[
|
|
"numpy",
|
|
"PyYAML",
|
|
f"torch-mlir=={TM_VERSION}",
|
|
]
|
|
+ backend_deps,
|
|
)
|