Chi_Liu
|
a0097a1ead
|
Add mlir_type for torch_model_list.csv (#1428)
- Enable stablehlo/tosa mlir output for torch model
- Add BERT stablehlo support
|
2023-05-15 10:23:54 -07:00 |
|
Chi_Liu
|
aa8ada9da9
|
Add support for torch to stablehlo and tosa in shark_importer (#1360)
|
2023-04-27 08:09:45 -07:00 |
|
Ean Garvey
|
481d0553d8
|
Remove unnecessary repro_dir / shark_tmp usage
|
2023-02-07 14:06:07 +05:30 |
|
yzhang93
|
fee73b0b63
|
Add SD model annotation on fly (#869)
* Add SD model annotation on fly
* Move tuned_compile_through_fx to utils
* Fix SD compilation flags
|
2023-01-26 11:46:36 -08:00 |
|
Prashant Kumar
|
7b1f04d121
|
Changes incorporating the recent torch_mlir compile api changes.
|
2022-11-15 15:25:37 +05:30 |
|
Ean Garvey
|
fd7baae548
|
Serialize torch-mlir CAPI module as bytecode instead of string. (#435)
* Serialize torch-mlir CAPI as bytecode instead of string.
* Minor fixes to MLIR data handling in SHARK python.
|
2022-10-27 14:37:15 -05:00 |
|
Chi_Liu
|
064aa3b1f4
|
Fix tmp dir bug (#285)
|
2022-08-22 15:00:35 -07:00 |
|
Prashant Kumar
|
02b1e7ac36
|
Update torch_mlir.compile API.
torch_mlir.compile API is updated and verified by compiling all the
torch models via generate_sharktank script.
|
2022-08-10 22:50:15 +05:30 |
|
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
|
1a861acbb2
|
Add the torch_mlir.compile API for static cases. (#174)
|
2022-07-06 09:21:17 -07:00 |
|
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 |
|
Prashant Kumar
|
4d3acff484
|
Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)
|
2022-06-16 07:28:00 -07:00 |
|
Stanley Winata
|
c70ab55717
|
Add saving and loading of mlir to run. (#89)
|
2022-06-05 08:42:40 -07:00 |
|
Prashant Kumar
|
15c99adab8
|
Refactor the whole SharkInference API.
1. SharkInference API is refactored to target any frontends or files,
i.e. pytorch, tensorflow, mhlo, linalg, tosa.
2. Refactor all torch APIs.
|
2022-05-26 18:55:47 +05:30 |
|
Prashant Kumar
|
4b63e0e04a
|
Add run_on_refbackend and gpu configs. (#52)
`run_on_refbackend` is added to run linalg_on_tensors_backend compiled
code on torch-mlir's refbackend. Also, added gpu configs and flags.
|
2022-05-23 14:18:11 -07:00 |
|
Prashant Kumar
|
37468f7bb8
|
Enable Shark Trainer APIs
|
2022-04-21 20:31:30 +05:30 |
|
monorimet
|
1f758a4278
|
Add in option to specify path to which module files are saved.
|
2022-04-15 18:11:56 -05:00 |
|
monorimet
|
692cd180f6
|
Add shark-runner command line options for saving modules as .mlir or .vmfb
|
2022-04-15 01:10:30 -05:00 |
|
Prashant Kumar
|
f9864b4ce1
|
Modify the directory structure to remove conflicts.
The directory structure is modified, also disabled the aot_module
unless the shark_trainer is deployed.
|
2022-04-04 21:23:33 +05:30 |
|