Anush Elangovan
4e4c34c717
fix release downloads
2022-09-13 15:00:47 -07:00
Daniel Garvey
64424877ac
No iree instal ( #294 )
...
* adds support to default to tuned model
currently setup for tf bert/resnet50
going to refactor test class to avoid having to
add an argument to 50+ files
* adds an option to avoid installing iree
useful when building iree from source
specify env variable NO_BACKEND=1
2022-08-25 15:02:28 -05:00
Ean Garvey
3514822cac
Improvements to pytest benchmarks. ( #267 )
...
* Add ONNX env var flags for venv setup.
* Setup arguments for ONNX benchmarking via pytest.
* Enable ONNX benchmarking on MiniLM via pytest (experimental)
* Fix sequence lengths to 128 for TF model creation and fix issue with benchmarks.
* Disable CI CPU benchmarks on A100, change some default args.
* add xfails for roberta TF model tests on GPU.
2022-08-17 02:29:48 -05:00
Ean Garvey
d556c0d6ef
Update nightly.yml to remove IMPORTER=1 in package validation. ( #226 )
...
* Update nightly.yml to fix wheel building issues.
* Update setup_venv.sh to only install torch+cu116 if IMPORTER=1 is set.
2022-07-31 20:39:59 -07:00
Ean Garvey
0ee515a7be
Fix GPU benchmarks on PyTorch model tests. ( #218 )
...
* Add CUDA_BENCHMARKS option to setup.venv to enable PyTorch benchmarks on CUDA.
* Fix PyTorch GPU benchmarks for tank models.
2022-07-29 15:17:55 -07:00
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
Anush Elangovan
e4dd179a73
Tweak requirements.txt
2022-06-12 02:01:29 -07:00
Anush Elangovan
879942d87f
Fix macOS importer reqs
2022-06-12 01:45:05 -07:00
Anush Elangovan
3c26c02a5e
Add OSX-importer requirements
2022-06-12 07:38:02 +00:00
Anush Elangovan
626482e46a
Refactor setup_venv.sh
...
IMPORTER=1 ./setup_venv.sh is required to get importer deps
Without importers venv is 1.1G, with it is 3.9G
2022-06-11 06:26:31 +00:00
Stanley Winata
8565be9b6b
ORT-HF Benchmark Integration ( #101 )
...
-Add HF Benchmarker class.
-Add sample to benchmark HF model.
Example:
```bash
python -m benchmarks.hf_model_benchmark --num_iterations=10 --model_name="microsoft/MiniLM-L12-H384-uncased"
```
2022-06-07 23:49:39 -07:00
Anush Elangovan
22aa0e51a7
Move macOS to 3.10
2022-06-05 10:36:31 -07:00
Stanley Winata
276dcf1441
Add benchmark for TF ( #87 )
...
-Refactor SharkBenchmarker to run TF
-Add example and test to benchmark TF
2022-06-03 22:10:27 -07:00
anush elangovan
af6dc91b83
Update vulkan check
2022-06-03 23:45:42 +00:00
anush elangovan
8fa5b39bd6
Add importer utils to linux build
2022-06-02 19:33:51 +00:00
Prashant Kumar
96577000a3
Remove functorch dependency.
2022-05-24 18:42:35 +05:30
anush elangovan
49bec49c53
prep for pypi
2022-05-16 05:44:18 +00:00
Stanley Winata
2117c1ab39
Install package to pip venv during setup_venv
...
On fresh pip venv, this commit will help install
the packages onto the newly create venv.
2022-05-15 15:00:17 -07:00
Anush Elangovan
c8aafae562
Update links to SHARK
2022-05-15 14:47:31 -07:00
Anush Elangovan
99bf24849d
Handle default venv
2022-05-15 13:58:30 -07:00
Anush Elangovan
6bf3f6864c
Use python3 instead of PYTHON3
2022-05-15 12:43:51 -07:00
Anush Elangovan
9e48a3525b
Add error checking and conda support
...
Committer: Anush Elangovan <anush@nod-labs.com >
2022-05-15 12:40:31 -07:00
powderluv
adb6b327dd
Update pyproject.toml for setup to work correctly
2022-05-01 21:02:22 -07:00
Prashant Kumar
bd212634c1
Add CI for testing of various models. ( #28 )
...
Added pytest CI for testing for Pytorch Inference and vision models.
2022-04-27 06:52:49 -07:00
powderluv
c708032fb2
M1 settings for venv
2022-04-26 22:00:09 -07:00
anush elangovan
35b1e33108
Setup packaging
2022-04-26 00:23:42 -07:00
Prashant Kumar
d9f47b59f2
Modify setup_venv to incorporate python version.
2022-04-18 17:14:58 +05:30
Prashant Kumar
5991a90284
Modify install.sh and update readme.
2022-04-18 11:29:15 +05:30
