415 Commits

Author SHA1 Message Date
powderluv
958e0100bf Update test-models.yml 2022-06-30 14:41:26 -07:00
powderluv
6ca726615e Update test-models.yml 2022-06-30 14:40:25 -07:00
powderluv
d1b398cb9b Update test-models.yml 2022-06-30 14:28:37 -07:00
powderluv
6b0bd7181f Update test-models.yml 2022-06-30 14:20:30 -07:00
powderluv
b24ce861d0 Update test-models.yml 2022-06-30 14:14:49 -07:00
powderluv
79e392d85f enable pip cache 2022-06-30 14:11:00 -07:00
powderluv
e57cee21a5 Delete shark/tmp directory 2022-06-30 14:03:35 -07:00
powderluv
a064a32755 Update test-models.yml 2022-06-30 13:53:19 -07:00
powderluv
24ba2bcdc8 Update test-models.yml 2022-06-30 13:50:54 -07:00
Anush Elangovan
a7435973d9 Fix black formatting 2022-06-30 20:42:02 +00:00
powderluv
cf4d363436 Update test-models.yml 2022-06-30 13:38:31 -07:00
powderluv
b012b9a5f7 Update test-models.yml 2022-06-30 13:37:16 -07:00
powderluv
9556060c27 Update test-models.yml 2022-06-30 13:23:20 -07:00
powderluv
1841fcc088 Update test-models.yml 2022-06-30 13:11:11 -07:00
powderluv
36e2f80fdb Update test-models.yml 2022-06-30 13:07:36 -07:00
powderluv
188ec5431f update to matrix builds 2022-06-30 13:04:02 -07:00
powderluv
7532a8db0c Update test-models.yml 2022-06-30 00:44:08 -07:00
powderluv
6ed957f5fb enable fast fail 2022-06-30 00:35:00 -07:00
powderluv
b01ffb5a22 delete black builder 2022-06-30 00:33:14 -07:00
Prashant Kumar
83855e7b08 Capture input information from mlir_graph and generate random inputs. 2022-06-29 22:51:53 +05:30
Chi_Liu
2adea76b8c Enable SharkImporter Tests (#165)
rewrite some tflite code from SharkImporter to tflite_utils.py
2022-06-29 10:11:48 -07:00
Chi_Liu
210d28fce9 Add more tflite shark_importer examples (#163) 2022-06-28 08:39:56 -07:00
Prashant Kumar
b07377cbfd Refactor the shark_runner shark_inference to only support mlir_modules.
1. The shark_inference is divided into shark_importer and
   shark_inference.
2. All the tank/pytorch tests have been updated.
2022-06-28 18:46:18 +05:30
Chi_Liu
44dce561e9 SharkImporter for tflite without forward and compile (#159) 2022-06-23 22:49:35 -07:00
powderluv
4ae9331a77 Update README.md 2022-06-23 22:10:56 -07:00
Ean Garvey
0a6bc6e17f Generate test-specific repro path for each TF model test. (#158)
Set TempFileSaver path directory to shark_args.repro_dir
2022-06-23 21:58:45 -07:00
Chi_Liu
fa0aaf63c2 Fix repo_dir cannot export/write mlir file bug (#157)
-Set repro_dir as a temporary dir within current working directory.
2022-06-23 18:37:16 -07:00
Chi_Liu
b7b2dfaa8d Fix shark_importer.py iree_utils import bug (#156) 2022-06-23 16:40:11 -07:00
powderluv
a7ca9b8d68 Update parser.py 2022-06-23 15:26:17 -07:00
powderluv
2c90ddde5b Update README.md 2022-06-23 15:17:14 -07:00
Ean Garvey
71f5cfcb30 Fix package management in tank for Nightly build. (#152)
Fix formatting.
2022-06-22 15:51:57 -07:00
Chi_Liu
a635b6fbef Add tf/torch/mhlo/tosa support for SharkDownloader (#151) 2022-06-22 11:25:34 -07:00
Prashant Kumar
e8aa105b2a Divide iree_utils and do module imports on function calls. 2022-06-22 14:17:33 +05:30
Ean Garvey
08eda2ce35 Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148) 2022-06-21 21:54:44 -07:00
Chi_Liu
206ba11cc8 Add SharkDownloader for end users (#147)
* Add SharkDownloader for user

* Change tank_url to gs://shark_tank
2022-06-21 16:42:23 -07:00
powderluv
6bc9ebad24 Update test-models.yml
ignore onnx benchmarks for now
2022-06-21 13:12:59 -07:00
Prashant Kumar
334d129443 Changes to minilm to make it one on one with tensorflow. 2022-06-21 22:08:59 +05:30
Anush Elangovan
28a620f0e2 Remove ONNX importer from requirements for now.
Seems to cause install conflicts.
2022-06-21 16:15:08 +00:00
Chi_Liu
af582925f2 Enable tosa.mlir as input for SharkImporter inference (#145)
* Change shark_importer to use tosa.mlir as tflite model input from  local gen_shark_tank
2022-06-20 23:15:14 -07:00
Chi_Liu
f6e9f2d571 Generate Shark Tank for tflite (#142)
generate shark_tank for tflite
2022-06-20 14:35:35 -07:00
Anush Elangovan
c66e285e2b Enable 64bit for models 2022-06-20 20:25:29 +00:00
Ean Garvey
7198f0db0e Add TempFileSaver pytest option to tank TF tests. (#144)
add/remove pytest marks to TF tests
2022-06-20 10:00:28 -07:00
Eliasj42
aae8754ace Fixed bug where backend would segfault under specific circumstances (#134)
* updated README

* added debug printouts

* added debug printouts

* fixed bug where backend would segfault on models with multiple outputs of different sizes
fixed formating issues

Co-authored-by: Elias Joseph <elias@nod-labs.com>
2022-06-20 09:56:58 -07:00
Ean Garvey
f1edf88c33 Revert "add/remove pytest marks to TF tests"
This reverts commit b522f933a9.
2022-06-20 16:06:40 +00:00
Ean Garvey
b522f933a9 add/remove pytest marks to TF tests 2022-06-20 15:23:28 +00:00
Ean Garvey
dbabb48582 Add TempFileSaver pytest option to tank TF tests. 2022-06-20 15:02:36 +00:00
Anush Elangovan
2b3fac7519 Vulkan xfails
https://github.com/google/iree/issues/9554
2022-06-18 09:39:09 +00:00
Anush Elangovan
de223e7824 Add xfail reason 2022-06-18 08:31:26 +00:00
powderluv
fd4df79573 Update README.md 2022-06-17 14:16:39 -07:00
powderluv
0093610054 Update test-models.yml
disable a100 cpu for now
2022-06-16 16:27:11 -07:00