Stanley Winata
c47218c972
WIP: Intel-GPU integration
2022-08-10 11:24:00 -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
