Ean Garvey
a90812133b
Enable pytests on Windows ( #901 )
2023-02-01 18:36:41 -06:00
Ean Garvey
72f29b67d5
Add Resnet50 fp16 variant to pytests. ( #760 )
2023-01-10 16:31:11 -08:00
Ean Garvey
40eea21863
Enable conv nchw-to-nhwc flag by default for most models + minor fixes ( #584 )
2022-12-07 16:24:02 -08:00
Ean Garvey
d2475ec169
Add mnasnet to torch models and minor fixes. ( #577 )
...
* Minor fixes to benchmark runner
* Add Mnasnet to tank.
2022-12-07 22:30:58 +05:30
Daniel Garvey
5df582e7e8
creates abstract test case class ( #333 )
2022-09-20 07:06:38 -07:00
Chi_Liu
17dba601c8
Add huggingface top5 image classification automodel ( #268 )
2022-08-22 15:05:38 -07:00
Ean Garvey
a8b021dc8d
Add benchmarks to MHLO miniLM and resnet50 and add dialect, num_iterations ( #264 )
2022-08-16 13:55:40 -05:00
powderluv
f7f24dc4d9
Revert "Add Debug log of torch_model_blacklist.txt ( #242 )" ( #249 )
...
This reverts commit 7023d556b5
2022-08-09 10:23:14 -07:00
Chi_Liu
7023d556b5
Add Debug log of torch_model_blacklist.txt ( #242 )
...
* Add debug log of torch_model_blacklist.txt
* Add make_fx for torch model
* Update torch_model_blacklists.txt
* Add some Xfails
2022-08-09 17:54:02 +05:30
Prashant Kumar
d934765b1d
Add mobilenet_v3_small torch model to the test_suite.
...
-- The model doesn't validate with the correct results on the GPU.
-- The model passes on CPU and levelzero.
-- The static version of the model gets stuck for vulkan.
2022-08-05 14:10:43 +05:30
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
Ean Garvey
c5ae01311d
Make PyTorch model tests importer-independent and add benchmarks.
2022-07-15 21:54:41 -05: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
Chi_Liu
d496c66d11
Torch support of gen_shark_tank ( #175 )
2022-07-06 22:42:15 -07:00
Prashant Kumar
e8aa105b2a
Divide iree_utils and do module imports on function calls.
2022-06-22 14:17:33 +05:30
Prashant Kumar
4d3acff484
Add tf masked lm models (17 models - 12 pass, 5 fail). ( #135 )
2022-06-16 07:28:00 -07:00
Ean Garvey
acdd1d1785
Update model testing utilities. ( #119 )
2022-06-13 02:16:30 -07:00
