mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
9956099516
* Use IREE tf tools to save .mlir modules when generating shark_tank. * Add option to pytest for enabling auto-updates to local shark tank. * xfail mobilenet torch on cpu, cuda and fix CI macos setup * Update test-models.yml to disable macos vulkan CI.
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 "NO_BACKEND" in os.environ.keys():
|
|
backend_deps = [
|
|
"iree-compiler>=20221022.190",
|
|
"iree-runtime>=20221022.190",
|
|
]
|
|
|
|
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",
|
|
"torch-mlir>=20221021.633",
|
|
]
|
|
+ backend_deps,
|
|
)
|