Ean Garvey
4b5d09fc6c
Add TF ResNet50 to tank tests. ( #261 )
...
* Add TensorFlow Resnet50 test to shark tank.
2022-08-12 09:20:43 -07:00
Ean Garvey
23619068eb
Disable passing of sm_arch to iree-compile CL args by default. ( #253 )
...
* Disable passing of sm_arch to iree-compile CL args by default.
* Fix formatting.
2022-08-10 01:19:24 -07:00
powderluv
1c4f5e0c34
Add M1 Max and Pro variants
2022-08-04 13:45:34 -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
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
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
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
Anush Elangovan
a7435973d9
Fix black formatting
2022-06-30 20:42:02 +00:00
Prashant Kumar
b07377cbfd
Refactor the shark_runner shark_inference to only support mlir_modules.
...
1. The shark_inference is divided into shark_importer and
shark_inference.
2. All the tank/pytorch tests have been updated.
2022-06-28 18:46:18 +05:30
Prashant Kumar
e8aa105b2a
Divide iree_utils and do module imports on function calls.
2022-06-22 14:17:33 +05:30
