Vivek Khandelwal
cc4fa96831
Remove op decomposition from the v_diffusion.py ( #210 )
...
The PyTorch decomposition for the op `aten.upsample_bilinear2d.vec`
is merged in the upstream repo and hence removed from this file.
2022-07-25 17:36:26 +05:30
Ean Garvey
921ccdc40b
Force TensorFlow to use CPU when generating model and golden values. ( #199 )
...
Split up gpu pytest runs temporarily
2022-07-22 20:21:16 -07:00
Prashant Kumar
49fc6d2f4b
Add all the tf models to gs::shark_tank. ( #207 )
...
-- Tensorflow models have been imported to shark_tank.
-- TF Tests have been updated to use the downloader.
2022-07-22 11:53:07 -05:00
Prashant Kumar
e62c531170
Update the torch_model for dynamic cases.
2022-07-22 00:21:35 +05:30
Prashant Kumar
3c2f8ef243
Generate shark_tank for tensorflow models.
...
Updated the generate_sharktank.py script to give tensorflow models.
2022-07-20 22:37:14 +05:30
Vivek Khandelwal
ec870f45fd
Merge pull request #197 from vivekkhandelwal1/v-diffusion
...
Add v-diffusion model
2022-07-20 21:41:48 +05:30
Vivek Khandelwal
0e2485d85b
Add v-diffusion model
2022-07-20 21:36:42 +05:30
Ean Garvey
6401687733
Update TF pytest marks and fix benchmark runner import. ( #196 )
2022-07-19 15:42:32 -05:00
Prashant Kumar
39c5f940c9
Add v_diffusion model into shark examples. ( #195 )
2022-07-19 09:51:10 -07:00
powderluv
9fc1c84cef
Switch to disable multiprocessing as it fails on macOS ( #194 )
2022-07-19 01:25:13 -07:00
Ean Garvey
fa6f19679e
Merge pull request #186 from nod-ai/benchmarks
...
Make PyTorch model tests importer-independent and add benchmarks.
2022-07-18 12:23:48 -05:00
Prashant Kumar
9105f5d54e
Use the downloaded folder instead of re-downloading.
...
shark_tank models.
2022-07-18 17:10:54 +05:30
Prashant Kumar
54a642e76a
Update the resnet50 example to use the shark_downloader.
...
The resnet50 example is updated to use the shark_downloader instead
of shark_importer and inference.
2022-07-18 13:41:38 +05:30
Ean Garvey
d047eff086
Update test-models.yml
2022-07-16 16:38:32 -05:00
monorimet
5da32601c0
Ensure CPU workflow job suite only collects CPU tests.
2022-07-16 13:50:36 -05:00
Ean Garvey
c5ae01311d
Make PyTorch model tests importer-independent and add benchmarks.
2022-07-15 21:54:41 -05:00
Prashant Kumar
1191f53c9d
Add the shark_downloader for the torch_models. ( #184 )
2022-07-14 13:41:43 -07:00
Chi_Liu
8434c67d96
Change tflite tests from sharkimporter -> sharkdownloader ( #182 )
...
* Change tflite test from sharkimporter -> sharkdownloader
* xfail all uint/int tflite sharkdownloader tests
2022-07-14 13:40:25 -07:00
Ean Garvey
79caf729f7
Merge pull request #169 from nod-ai/tf-tests
...
Refactor TF tests for importer/runner split
2022-07-14 14:58:18 -05:00
monorimet
b647764b9a
Update requirements.txt
2022-07-14 13:07:46 -05:00
monorimet
08448a013f
Refactor TF tests for importer split, update pytorch tests.
2022-07-14 12:59:21 -05:00
gpetters94
69f169be1b
Update model status ( #181 )
2022-07-12 20:38:43 -07:00
Prashant Kumar
0dcf387089
Add shark_importer for torch_models. ( #183 )
...
All the torch_models are imported to gs::shark_tank.
Scripts have been updated.
2022-07-12 20:38:19 -07:00
Prashant Kumar
2e22d0b690
Bug fix: Pass the device attribute appropriately.
...
Previously the device attribute was not passed and device was
hardcoded to "cpu". So every tests were running on cpu.
2022-07-09 01:56:17 +05:30
Chi_Liu
b98c9ee484
torch fix in gen_shark_tank ( #180 )
2022-07-08 10:13:58 -07:00
Prashant Kumar
fa7ee7e099
Update pytorch tests to support vulkan and cuda.
...
All the model validation pass except distilbert which is failing
in torch-mlir lowering. Also, added the mobilebert-uncased model to
the torch test suite.
2022-07-08 14:40:13 +05:30
Prashant Kumar
9cc92d0e7d
Update the README.md
...
Update the shark_importer and shark_inference APIs.
2022-07-07 21:08:32 +05:30
Prashant Kumar
2e5cb4ba76
Add resnest model to the shark_inference examples list.
2022-07-07 19:51:53 +05:30
gpetters94
f49a2c3df4
Update README with pending models supported by groups ( #176 )
2022-07-07 00:01:20 -07:00
Chi_Liu
d496c66d11
Torch support of gen_shark_tank ( #175 )
2022-07-06 22:42:15 -07:00
Prashant Kumar
1a861acbb2
Add the torch_mlir.compile API for static cases. ( #174 )
2022-07-06 09:21:17 -07:00
Chi_Liu
1cad50d521
generate shark tank for tflite ( #173 )
...
* Add gen_shark_tank support tflite
* gen_shark_tank.py use SharkImporter to save model
2022-07-05 23:11:19 -07:00
powderluv
c351bb50b6
Update test-models.yml
2022-07-03 13:56:45 -07:00
powderluv
ca36d40f0a
Update test-models.yml
2022-07-01 12:27:49 -07:00
Chi_Liu
4e9e35a9ff
Fix mobilebert return bug ( #171 )
2022-07-01 05:46:18 -07:00
Chi_Liu
41a8cbb5b6
Fix deeplab&mobilebert tflite test bug ( #170 )
2022-06-30 21:42:14 -07:00
powderluv
cc11a71ec8
Update test-models.yml
2022-06-30 18:55:38 -07:00
powderluv
1d1a1b131f
Update test-models.yml
2022-06-30 18:54:50 -07:00
powderluv
84a2969381
Update test-models.yml
2022-06-30 15:57:07 -07:00
Chi_Liu
babd3d002b
Merge pull request #164 from AmosLewis/move
...
Rewrite&Move tflite examples to up tank dir
2022-06-30 15:47:15 -07:00
Chi Liu
c1cde2e252
Add xfail with sys.platform == "darwin" for mac
2022-06-30 15:17:59 -07:00
Chi Liu
638f982c94
Add Xfail for tflites exampls
...
Know Macos iree-tflite install error
2022-06-30 15:02:05 -07:00
Chi Liu
84978cf3c5
Fix tflite examples fail pytest
2022-06-30 14:57:33 -07:00
Chi Liu
96dd08cca4
Fix load json input bug in SharkDownloader albert test
2022-06-30 14:57:33 -07:00
Chi Liu
06a45d9025
Rewrite tflite SharkImporter tests to compatible to most recent SharkImporter
2022-06-30 14:57:33 -07:00
Chi Liu
193a735865
Move tflite examples to up dir
2022-06-30 14:57:33 -07:00
powderluv
8199ea15e8
Update test-models.yml
2022-06-30 14:42:41 -07:00
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
