remove shark 1.0 tests, add support for 2.0 llm

* add support for external weights

* add tests and edit deps
This commit is contained in:
Daniel Garvey
2023-12-14 21:44:37 -06:00
committed by GitHub
parent f692a012e1
commit ebfcfec338
16 changed files with 377 additions and 576 deletions

View File

@@ -9,11 +9,6 @@ with open("README.md", "r", encoding="utf-8") as fh:
PACKAGE_VERSION = os.environ.get("SHARK_PACKAGE_VERSION") or "0.0.5"
backend_deps = []
if "NO_BACKEND" in os.environ.keys():
backend_deps = [
"iree-compiler>=20221022.190",
"iree-runtime>=20221022.190",
]
setup(
name="nodai-SHARK",
@@ -39,7 +34,5 @@ setup(
install_requires=[
"numpy",
"PyYAML",
"torch-mlir",
]
+ backend_deps,
)