mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 22:38:01 -05:00
64424877ac
* adds support to default to tuned model currently setup for tf bert/resnet50 going to refactor test class to avoid having to add an argument to 50+ files * adds an option to avoid installing iree useful when building iree from source specify env variable NO_BACKEND=1
44 lines
1.2 KiB
Python
44 lines
1.2 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()
|
|
|
|
PACKAGE_VERSION = os.environ.get("SHARK_PACKAGE_VERSION") or "0.0.4"
|
|
backend_deps = []
|
|
if os.environ["NO_BACKEND"] is None:
|
|
backend_deps = [
|
|
"iree-compiler>=20220427.13",
|
|
"iree-runtime>=20220427.13",
|
|
]
|
|
|
|
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.7",
|
|
install_requires=[
|
|
"numpy",
|
|
"PyYAML",
|
|
"torch-mlir>=20220428.420",
|
|
]
|
|
+ backend_deps,
|
|
)
|