Ean Garvey
905d0103ff
Revert "Re-enable SD tunings without matmuls. ( #1976 )" ( #1979 )
...
This reverts commit 70817bb50a
2023-11-17 23:44:33 +05:30
Ean Garvey
70817bb50a
Re-enable SD tunings without matmuls. ( #1976 )
2023-11-15 20:42:53 -06:00
yzhang93
47b346d428
Modify the lowering config format for SPIRVMatmulPromoteVectorize pipeline ( #1471 )
2023-05-29 21:53:48 -07:00
Ean Garvey
a9bae00606
Fix vulkan device selection at compile time and adapt to IREE python changes. ( #1407 )
...
* Add support for vulkan device selection at compile time.
* Don't convert device ID to int and fix .exe imports
2023-05-12 23:31:50 -07:00
Ean Garvey
6c95ac0f37
Revert dialect registration in model annotator ( #1406 )
...
Matches https://github.com/nod-ai/SHARK-Runtime/pull/58
2023-05-09 11:50:19 -07:00
Ean Garvey
cdf2d61d53
Remove imports from iree.compiler.transforms from model annotator. ( #1392 )
2023-05-04 20:40:19 -07:00
yzhang93
9644e78545
Fix CUDA tuned model annotation ( #880 )
2023-01-27 11:35:18 -08:00
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
yzhang93
782b449c71
Add script to auto annotate SD models and variants ( #751 )
...
* Add script to auto annotate SD models and variants
* Add model config files
* Add script to auto annotate SD models and variants
* Add model config files
* Move config files to shark_tank
2023-01-04 15:53:10 -08:00
yzhang93
1595254eab
Modify model annotation tool to walk through ops by shape ( #692 )
2022-12-21 10:46:30 -08:00
yzhang93
0754c6ea20
Update model annotation to take vulkan configs ( #495 )
...
Co-authored-by: vivian <vivian@nod-labs.com >
2022-11-17 14:34:17 -08:00
yzhang93
587d74b449
Update model annotation tool ( #361 )
...
Usage:
with create_context() as ctx:
module = model_annotation(ctx, input_contents=..., config_path=..., search_op=...)
Example:
The example is to annotate the minilm model with GPU config files.
python model_annotation.py /nodclouddata/vivian/minilm_model/model.mlir /nodclouddata/vivian/minilm_model/model_config.json
2022-09-23 15:44:51 -07: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
Anush Elangovan
a7435973d9
Fix black formatting
2022-06-30 20:42:02 +00:00
Prashant Kumar
e8aa105b2a
Divide iree_utils and do module imports on function calls.
2022-06-22 14:17:33 +05:30
Prashant Kumar
9d2ce33bd1
yapf format python files.
...
Please use `yapf -i --style .style.yapf shark/*.py` to format python
files.
2022-06-06 14:34:59 +05:30
yzhang93
16c50cac15
Add the option to use tuned model in shark_runner ( #79 )
2022-06-02 18:25:33 -07:00
