mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
Revert "Move most xfails to entries in tank/all_models.csv and temporarily remove multiprocessing and TF gpu support. (#602)" (#622)
This reverts commit fe618811ee.
This commit is contained in:
12
setup.py
12
setup.py
@@ -6,18 +6,12 @@ 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}",
|
||||
"iree-compiler>=20221022.190",
|
||||
"iree-runtime>=20221022.190",
|
||||
]
|
||||
|
||||
setup(
|
||||
@@ -43,7 +37,7 @@ setup(
|
||||
install_requires=[
|
||||
"numpy",
|
||||
"PyYAML",
|
||||
f"torch-mlir=={TM_VERSION}",
|
||||
"torch-mlir>=20221021.633",
|
||||
]
|
||||
+ backend_deps,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user