7 Commits

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