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:
powderluv
2022-12-13 21:49:46 -08:00
committed by GitHub
parent fe618811ee
commit d946cffabc
13 changed files with 168 additions and 134 deletions

View File

@@ -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,
)