Stanley Winata
c47218c972
WIP: Intel-GPU integration
2022-08-10 11:24:00 -07:00
powderluv
c6b4ad1e26
Update test-models.yml
2022-08-02 13:19:10 -07:00
Phaneesh Barwaria
198755788c
Check for MacStudio Target Value ( #233 )
...
* Identify Apple M2 for Vulkan
* Check for MacOS Target
2022-08-02 13:18:36 -07:00
Ean Garvey
c1a14a8db1
Update wide_resnet50_2_test.py ( #229 )
...
Removed old xfails.
2022-08-01 20:38:53 -05:00
Chi_Liu
0e42c19f33
Test tflite quantization_info fail on cloud ( #228 )
2022-08-01 18:21:04 -07:00
Chi_Liu
0c31bb82cd
Add hash of mlir for tf/tflite ( #225 )
2022-08-01 10:40:49 -07:00
Prashant Kumar
315ec72984
Add check for hash.npy in the models folder.
2022-08-01 22:40:25 +05:30
Ean Garvey
d556c0d6ef
Update nightly.yml to remove IMPORTER=1 in package validation. ( #226 )
...
* Update nightly.yml to fix wheel building issues.
* Update setup_venv.sh to only install torch+cu116 if IMPORTER=1 is set.
2022-07-31 20:39:59 -07:00
Ean Garvey
0ee515a7be
Fix GPU benchmarks on PyTorch model tests. ( #218 )
...
* Add CUDA_BENCHMARKS option to setup.venv to enable PyTorch benchmarks on CUDA.
* Fix PyTorch GPU benchmarks for tank models.
2022-07-29 15:17:55 -07:00
Daniel Garvey
4c0deb9899
remove gpu script ( #220 )
...
No longer needed after fixes in iree
2022-07-29 15:20:03 -05:00
Chi_Liu
94d9275515
Move tf tests to tank/ ( #223 )
2022-07-29 09:34:50 -07:00
Chi_Liu
c1cb7bb3fd
Rename tank tflite/torch model dir ( #219 )
2022-07-29 01:56:05 -05:00
Prashant Kumar
ff20ddeb97
Add hash of the mlir for checking upstream changes. ( #217 )
...
-- hash of the mlir files have been added in the shark_tank. (If there
is upstream changes, the model will be downloaded otherwise not).
-- The models have been placed in `~/.local/shark_tank/` folder as
opposed to the current folder not to redownload the files again.
2022-07-28 21:54:50 -05:00
Chi_Liu
2c7d879a4e
Add tf image classification auto model tests ( #216 )
2022-07-27 18:38:13 -07:00
Prashant Kumar
4e5c592094
Enable --devices to pass through CLI.
2022-07-27 22:13:52 +05:30
Chi_Liu
af4257d05f
Add tf image classification auto model ( #213 )
2022-07-26 23:18:42 -07:00
Chi_Liu
dc1a283ab7
Remove tf package dep for SharkDownloader tflite tests ( #212 )
2022-07-26 12:41:12 -07:00
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