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