Ean Garvey
9697981004
Pipe through a debug option to iree compile utils. ( #1796 )
...
* Update compile_utils.py
* Pipe through a flag to toggle debug options in compile utils.
* Update SharkLLMBase.py
2023-08-25 07:11:11 -07:00
Ean Garvey
07f6f4a2f7
Add a short README for the OPT examples and small tweaks. ( #1793 )
...
* Small changes to OPT example.
* Update opt README.
* Add a few modes to batch script.
* Update README.md
2023-08-24 17:26:11 -07:00
Ean Garvey
8e3860c9e6
Remove flags that are default in upstream IREE ( #1785 )
...
* Remove index bits flags now set by default
* Update shark_studio_imports.py
2023-08-24 11:57:54 -05:00
Sungsoon Cho
79075a1a07
Opt perf ( #1786 )
...
* Define command line args, model-name, max-seq-len, platform, etc.
* Add usage example.
* Add opt_perf_comparision_batch.py.
* Use shlex instead.
2023-08-24 08:33:12 -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
6e6c90f62b
Pin torch-mlir and use local-task in OPT. ( #1592 )
2023-06-23 19:17:05 -07: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
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
218ed78c40
Change instances of input_type='mhlo' to 'auto' ( #1482 )
2023-06-02 16:43:47 -05: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
Thomas
807947c0c8
Remove deprecated cli option iree-hal-cuda-disable-loop-nounroll-wa ( #1235 )
2023-03-22 12:05:15 -05:00
Boian Petkantchin
815e23a0b8
Update iree-compile flags --iree-llvm-xxx -> --iree-llvmcpu-xxx ( #1164 )
2023-03-09 11:31:50 -08:00
Ean Garvey
a504903dd5
Fix formatting issues. ( #903 )
2023-02-01 09:12:45 +05:30
Daniel Garvey
8859853b41
Revert "Revert "find gsutil on linux ( #557 )" ( #560 )" ( #561 )
...
This reverts commit 3c46021102
2022-12-05 20:27:43 -08:00
Daniel Garvey
3c46021102
Revert "find gsutil on linux ( #557 )" ( #560 )
...
This reverts commit bba8646669
2022-12-05 21:53:47 -06:00
Daniel Garvey
bba8646669
find gsutil on linux ( #557 )
...
* find gsutil on linux
* cleaned up downloader and ditched gsutil
Co-authored-by: dan <dan@nod-labs.com >
2022-12-05 19:03:48 -08:00
Ean Garvey
45677c1e23
Install torch version required by torch-mlir when setting up importer venv. ( #486 )
2022-11-14 14:01:01 +05:30
mariecwhite
ec461a4456
Enable XLA compiler for TF models ( #484 )
2022-11-13 20:10:47 -08:00
Ean Garvey
f3bde3c7fc
Cleanup tank directory and move instructions to tank/README.md ( #401 )
2022-10-13 12:20:02 -05:00
