Prashant Kumar
|
4e5c592094
|
Enable --devices to pass through CLI.
|
2022-07-27 22:13:52 +05:30 |
|
Ean Garvey
|
6401687733
|
Update TF pytest marks and fix benchmark runner import. (#196)
|
2022-07-19 15:42:32 -05: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 |
|
Anush Elangovan
|
a7435973d9
|
Fix black formatting
|
2022-06-30 20:42:02 +00:00 |
|
Prashant Kumar
|
83855e7b08
|
Capture input information from mlir_graph and generate random inputs.
|
2022-06-29 22:51:53 +05:30 |
|
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 |
|
Ean Garvey
|
08eda2ce35
|
Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148)
|
2022-06-21 21:54:44 -07:00 |
|
Chi_Liu
|
af582925f2
|
Enable tosa.mlir as input for SharkImporter inference (#145)
* Change shark_importer to use tosa.mlir as tflite model input from local gen_shark_tank
|
2022-06-20 23:15:14 -07:00 |
|
Prashant Kumar
|
4d3acff484
|
Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)
|
2022-06-16 07:28:00 -07:00 |
|
Prashant Kumar
|
380b0b7c54
|
Add seq_classification shark module to run seq_classification
on hugging face supported models.
|
2022-06-14 22:36:19 +05:30 |
|
Stanley Winata
|
e48642f8a5
|
Update benchmark command to ToM and Add to test (#125)
-Update benchmark_cl to latest benchmark_module API
-tensorflow OOM issue.
|
2022-06-13 23:57:05 -07:00 |
|
Ean Garvey
|
cbd131d588
|
Refactor testing framework. (#117)
|
2022-06-12 20:15:30 -07:00 |
|
Chi_Liu
|
8640c3ebfe
|
Add shark_importer tflite module and albert_shark_test example (#110)
|
2022-06-09 06:17:22 -07:00 |
|
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 |
|
Prashant Kumar
|
cee02f6a61
|
Add frontend error checks.
|
2022-05-27 15:45:45 +05:30 |
|
Prashant Kumar
|
e5517f63f5
|
Add the refactored training module.
|
2022-05-27 14:21:34 +05:30 |
|
stanley
|
91867e1fc2
|
Add Shark Benchmark
-Introduce SharkBenchmark that bench models on regular torch, shark-py, and shark-c.
-Integrate iree-benchmark-module into Shark.
|
2022-05-27 00:07:31 +00: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 |
|