Commit Graph

1743 Commits

Author SHA1 Message Date
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
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