github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update benchmark command to ToM and Add to test (#125)

Browse Source 
-Update benchmark_cl to latest benchmark_module API
-tensorflow OOM issue.
This commit is contained in:
Stanley Winata
2022-06-13 23:57:05 -07:00
committed by GitHub
parent 61b19ce091
commit e48642f8a5
8 changed files with 37 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@@ -4,7 +4,7 @@ requires = [
"wheel",
"packaging",
"numpy",
"numpy==1.22.4",
"torch-mlir>=20220428.420",
"iree-compiler>=20220427.13",
"iree-runtime>=20220427.13",