Ean Garvey
9c8cbaf498
Add support for ROCM (Windows) in Studio + compile utils ( #1770 )
...
* WIP: MSVC ROCM support for SHARK Studio
* Make get_iree_rocm_args platform-agnostic.
* Update stable_args.py
* Update rocm arg handling in SD utils
* Guard quantization imports.
Co-authored-by: jam https://github.com/jammm
2023-08-25 20:56:05 -07:00
Ean Garvey
343dfd901c
Update SHARK-Runtime links to SRT ( #1765 )
...
* Update nightly.yml
* Update setup_venv.ps1
* Update CMakeLists.txt
* Update shark_iree_profiling.md
* Update setup_venv.sh
* Update README.md
* Update .gitmodules
* Update CMakeLists.txt
* Update README.md
* fix signtool flags
* Update nightly.yml
* Update benchmark_utils.py
* uncomment tkinter launch
2023-08-15 12:40:44 -07:00
Ean Garvey
c96571855a
prevents recompiles for cuda benchmarks + update benchmark_module path ( #1759 )
...
* xfail resnet50_fp16
* Fix cuda benchmarks and prevent recompilation.
2023-08-14 15:30:32 -05:00
Phoenix Meadowlark
d319f4684e
Add peak memory reporting for IREE, TF and PyTorch ( #1216 )
2023-03-20 15:40:49 -05: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
yzhang93
7cf31a6ae4
Fix iree-benchmark flag names ( #952 )
2023-02-04 22:24:18 -08:00
Ean Garvey
a90812133b
Enable pytests on Windows ( #901 )
2023-02-01 18:36:41 -06:00
Ean Garvey
1699db79b5
Disable SHARK-Runtime flags if USE_IREE=1 specified during setup. ( #588 )
...
* Disable SHARK-Runtime flags if USE_IREE=1 specified during setup.
* Update setup_venv.sh
* Autodetect cpu count for runtime flags.
2022-12-08 02:31:31 -06:00
Phaneesh Barwaria
749a2c2dec
add support for choosing vulkan device ( #439 )
2022-11-12 14:00:41 -08:00
yzhang93
9a86e5c476
Fix dispatch benchmarking tool ( #460 )
2022-11-08 09:37:12 -08:00
Eliasj42
7f37599a60
Added a dispatch benchmarking tool ( #441 )
...
To produce benchmarks of individual dispatches, you can add --dispatch_benchmarks=All --dispatch_benchmarks_dir=<output_dir> to your command line argument.
Co-authored-by: Elias Joseph <elias@nod-labs.com >
2022-10-28 14:31:03 -07:00
Ean Garvey
4b5d09fc6c
Add TF ResNet50 to tank tests. ( #261 )
...
* Add TensorFlow Resnet50 test to shark tank.
2022-08-12 09:20:43 -07: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
e8aa105b2a
Divide iree_utils and do module imports on function calls.
2022-06-22 14:17:33 +05:30
