github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 22:07:55 -05:00
Code Issues Packages Projects Releases Wiki Activity
535 Commits 66 Branches 705 Tags
995609951697f62b55b54d89ba47807c75a37339
Commit Graph

37 Commits

Author SHA1 Message Date
Ean Garvey
9956099516 Add pytest option for updating tank and fix save_mlir function. (#413) 
* Use IREE tf tools to save .mlir modules when generating shark_tank.

* Add option to pytest for enabling auto-updates to local shark tank.

* xfail mobilenet torch on cpu, cuda and fix CI macos setup

* Update test-models.yml to disable macos vulkan CI.
 2022-10-25 21:29:18 +05:30
anush elangovan
a3f733490c Force update of packages 
Pickup tools from upstream IREE
 2022-10-19 05:20:53 +00:00
anush elangovan
8a11f138d1 Update SHARK-Runtime releases page 2022-10-19 05:06:36 +00:00
Ean Garvey
13bae2538a Update URL for IREE compiler/runtime install (#397) 
* Update URL for IREE compiler/runtime install

* Update gh-pages-releases.yml

* Update test_models.py

* Update assets path
 2022-10-12 15:47:11 -05:00
Daniel Garvey
d38e37bd99 seperate importer and benchmark deps (#393) 2022-10-08 23:31:20 -05:00
Ean Garvey
006adf8746 Fix issue with FASTAPI pip install. (#382) 2022-10-01 14:55:24 -05:00
Ean Garvey
56f8a0d85a Update torch-mlir releases page in setup_venv.sh (#374) 
* Update README.md

* Update setup_venv.sh
 2022-09-28 11:07:44 -07:00
Anush Elangovan
e304041574 Remove redundant {} 2022-09-13 16:12:35 -07:00
Anush Elangovan
1776c55e73 Fix torch-mlir download URL 2022-09-13 16:07:25 -07:00
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