Ean Garvey
a9f95a218b
Remove SD from all_models.csv ( #1706 )
...
Removes SD from pytests as it has its own test suite.
2023-08-05 15:55:52 -05:00
Ean Garvey
dd738a0e02
small changes to opt_perf_comparison.py ( #1670 )
...
* Use longer prompts for OPT comparison script
* small tweaks
2023-07-19 06:26:50 -07:00
Ean Garvey
7cf19230e2
add perf comparison script for opt. ( #1650 )
2023-07-13 13:29:48 -05:00
Ean Garvey
7dcbbde523
Xfail models for data tiling flag changes ( #1624 )
2023-07-06 06:57:17 -07:00
Ean Garvey
6e6c90f62b
Pin torch-mlir and use local-task in OPT. ( #1592 )
2023-06-23 19:17:05 -07:00
Chi_Liu
f60ab1f4fa
Add Deberta to stablehlo in shark tank ( #1545 )
2023-06-16 13:24:44 -07:00
Ean Garvey
eb6d11cfed
Change mlir dialects for tf tests to stablehlo. ( #1535 )
...
* Change mlir dialects for tf tests to stablehlo
* Update shark_runner.py
2023-06-14 10:43:49 -07:00
Ean Garvey
49b3ecdbca
(pytest) don't run redundant tests in cpu suite ( #1532 )
2023-06-13 22:40:33 -05:00
Ean Garvey
f53e3594c3
OPT Refactor ( #1516 )
...
* Change script to 1.3b model and add pytorch comparison
* fix CLI command
* Match OPT transformers model updates + numerics against latest version
* Cleanup OPT sentence completion script.
* Fix formatting and add standalone validation scripts.
* Add minimal OPT wrapper and example with import_with_fx
* Rename OPT full model wrapper.
* Cleanup test scripts for OPT.
2023-06-13 22:40:07 -05:00
Ean Garvey
5562d1dfda
Fix xfails for cpu pytest cases ( #1527 )
...
Adding cpu-sync and cpu-task device configs was allowing respective tests to bypass the xfail conditional for cpu pytests marked in tank/all_models.csv. This commit updates the conditional to xfail those cases for cpu-sync and cpu-task as well.
2023-06-13 17:01:51 -07:00
Ean Garvey
2c3485ca3e
Add standalone OPT sentence completion script. ( #1506 )
2023-06-07 10:58:03 -07:00
Ean Garvey
a07fe80530
Update OPT, ResNet example scripts. ( #1492 )
...
* Update API in OPT example.
* fix resnet50 script
* Add OPT1.3b test script.
2023-06-05 20:19:35 -07:00
Ean Garvey
4151ec3a8f
(pytest) tag efficientnet, mobilenet as xfails on vulkan ( #1488 )
2023-06-04 23:22:32 -07:00
Anush Elangovan
7ef1bea953
XFAIL some macos tests
2023-06-04 15:27:03 -07:00
Chi_Liu
ad89bb1413
Add distilgpt2 to stablehlo in shark tank ( #1481 )
2023-06-02 16:44:46 -05:00
Ean Garvey
218ed78c40
Change instances of input_type='mhlo' to 'auto' ( #1482 )
2023-06-02 16:43:47 -05:00
Chi_Liu
ba8eddb12f
Add GPT3/OPT to Stablehlo in shark tank ( #1468 )
...
Co-authored-by: AmosLewis <Amos_Lewsi@foxmail.com >
Co-authored-by: Ean Garvey <87458719+monorimet@users.noreply.github.com >
2023-05-29 21:58:39 -07:00
Ean Garvey
3cddd01b10
Update OPT tokenizer and xfail a few more large tests on macos CI ( #1459 )
...
* Update opt_torch_test.py
* Update all_models.csv
2023-05-23 14:36:57 -07:00
Chi_Liu
64c2b2d96b
Add gpt2 to stablehlo support in shark tank ( #1447 )
...
- Add torch decomposition support when generating shark tank
- Add gpt2 stablehlo
2023-05-22 10:45:51 -07:00
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
Ean Garvey
4c07e47e8c
Specify a few models for expected failure on CUDA CI. ( #1430 )
2023-05-12 17:03:37 -05:00
Ean Garvey
6c14847d1f
xfail some large tests on macOS builder and switch to hash updates. ( #1341 )
...
* Update test-models.yml
* Disable large tests on macOS builder
2023-05-04 19:47:03 -05:00
Daniel Garvey
7330729c92
enable sd pytest ( #1322 )
2023-04-19 22:11:30 -05:00
Ean Garvey
2191fc8952
Separate pytest benchmark modes and fix model updates for SHARK downloader / pytest. ( #1264 )
...
* Only xfail windows models in CI
* downloader: make model updates more robust.
* Separate baseline and native benchmarks in pytest.
* Fix native benchmarks
* Fix torchvision model utils.
2023-04-03 08:24:21 -07:00
Ean Garvey
d6f740b998
allow pytest to retry getting model artifacts + disable autotuning for pytorch benchmarks ( #1257 )
...
* Adds a few xfails to enable macOS builder
* Convert string batch sizes to ints where needed.
* allow pytest to retry getting model artifacts
* Reduce attempts and add assert msg.
2023-03-28 23:38:45 -05:00
Ean Garvey
96b1560da5
Make batch size configurable via pytest and fix sharktank generation. ( #1227 )
...
* Fix sharktank generation and add batch_size pytest option for torch.
* Disable torch dynamo until py3.11 supported
* Compile torchmodel without dynamo if torch.compile fails
* Use release versions of TF/Keras for importer.
* Pin torchvision and remove debug prints.
* Remove duplicates from torch model list.
* Update generate_sharktank.py
* xfail a few models that fail sharktank generation/ numerics
2023-03-28 14:33:39 -05:00
Ean Garvey
7619e76bbd
Disable and xfail some models that fail validation/compilation. ( #1251 )
...
* Rollback T5 models for torch as the inputs give some issues that aren't trivial to resolve
* xfail efficientnet-b0 on torch+cuda -- see CUDA requesting shared memory size larger than allowed size openxla/iree#12771
2023-03-27 12:42:53 -05:00
Daniel Garvey
9267eadbfa
disable openjourney gen for nightly ( #1249 )
2023-03-27 11:55:34 -05:00
Boian Petkantchin
4fac46f7bb
In models testing fix paths to be relative to the script dir not cwd ( #1128 )
...
authored-by: Boian Petkantchin <boian@nod-labs.com >
2023-03-22 15:26:52 -05:00
Thomas
807947c0c8
Remove deprecated cli option iree-hal-cuda-disable-loop-nounroll-wa ( #1235 )
2023-03-22 12:05:15 -05:00
Ean Garvey
e3c96a2b9d
Move sentencepiece to importer requirements. ( #1218 )
2023-03-21 00:39:57 -05:00
mariecwhite
d105246b9c
Fix t5 models
2023-03-21 10:39:59 +11:00
mariecwhite
90c958bca2
Add T5-base and T5-large Torch and TF Models ( #1116 )
2023-03-20 17:32:50 -05:00
mariecwhite
f99903e023
Add EfficientNet B0 and B7 Torch and TF models
2023-03-21 09:22:05 +11:00
mariecwhite
c6f44ef1b3
Add EfficientNet B0 and B7 Torch and TF models
2023-03-21 09:14:45 +11:00
mariecwhite
8dcd4d5aeb
Make batch size configurable
2023-03-20 18:03:17 -04:00
Ean Garvey
54d7b6d83e
Generate model artifacts in pytests if they don't exist in the cloud. ( #1121 )
...
* Add gen_shark_files fn to shark_downloader for OTF artifact generation
* add generate_sharktank as a tank/ python module.
* Fix some paths in tank generation.
2023-03-20 12:13:19 -05:00
Daniel Garvey
62b5a9fd49
generate sharktank for apps dir ( #966 )
...
* merge confix resolution
* add support to other scripts
---------
Co-authored-by: dan <dan@nod-labs.com >
2023-03-13 10:54:15 -07:00
Boian Petkantchin
815e23a0b8
Update iree-compile flags --iree-llvm-xxx -> --iree-llvmcpu-xxx ( #1164 )
2023-03-09 11:31:50 -08:00
Daniel Garvey
812152485d
temporarily xfail tiny convnext macos ( #1142 )
2023-03-03 13:30:56 -06:00
Ean Garvey
116b128802
Use nightly shark_tank for test-models ( #1133 )
...
* Use nightly shark_tank for test-models
* Update all_models.csv
2023-03-02 12:33:36 -06:00
Daniel Garvey
23f1468cc6
disable most models on windows pytest ( #1125 )
2023-03-02 01:37:50 -06:00
Daniel Garvey
b1113ab551
disable benchmark on windows for pytest ( #1100 )
2023-02-28 18:10:29 -06:00
Ean Garvey
6d2a485264
Add --benchmark_dispatches option to pytest. ( #800 )
...
* Add --benchmark_dispatches option to pytest.
* Update README.md and fix filepath for dispatch benchmarks
2023-02-19 12:16:18 -06:00
Daniel Garvey
5b204bee86
temporarily xfail microsoft resnet50 ( #1037 )
...
Co-authored-by: dan <dan@nod-labs.com >
2023-02-16 16:14:51 -06:00
mariecwhite
c3a4fdcbfc
Add bert-large-uncased TF model
2023-02-15 21:42:44 -08:00
mariecwhite
aac5de5b02
Add bert-large-uncased Torch model
2023-02-15 21:25:32 -08:00
Ean Garvey
1eb9436836
Fix generate_sharktank args.
2023-02-07 14:06:07 +05:30
Ean Garvey
9604d9ce81
make --update_tank update only if hash mismatch
2023-02-07 14:06:07 +05:30
Ean Garvey
481d0553d8
Remove unnecessary repro_dir / shark_tmp usage
2023-02-07 14:06:07 +05:30