dan
6eea111f5e
bert training wip
2022-11-17 17:24:46 +00:00
powderluv
a63755bc24
Correct spelling
2022-10-11 01:53:55 -07:00
Quinn Dawkins
d93d0783a8
Add script for tensorflow stable diffusion ( #391 )
2022-10-10 12:01:49 -04:00
Daniel Garvey
d38e37bd99
seperate importer and benchmark deps ( #393 )
2022-10-08 23:31:20 -05:00
Ean Garvey
3618fb3ada
Move old test scripts out of base tank directory and add xfails. ( #389 )
2022-10-07 16:02:46 -07:00
Vivek Khandelwal
70a29b03e0
Add FP16 Resnet50 script
2022-10-06 21:56:43 +05:30
Ean Garvey
006adf8746
Fix issue with FASTAPI pip install. ( #382 )
2022-10-01 14:55:24 -05:00
Quinn Dawkins
33b53e7caf
Add flag for specifying the vae mlir file location in stable diffusion ( #381 )
2022-09-30 00:37:58 -04:00
Daniel Garvey
c54815de17
edit assets path ( #376 )
2022-09-28 16:42:36 -05:00
Gaurav Shukla
0013fb0753
[WEB] Add shark-web logging
...
1. This commit adds support to display logs in the shark-web.
2. It also adds nod logo in the home page.
3. Stable-diffusion outputs are being saved now.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2022-09-29 01:20:42 +05:30
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
Ean Garvey
9035a2eed3
Add --local_tank_cache flag and update requirements. ( #368 )
...
* Add --local_tank_cache flag and update requirements.
* Update requirements-importer.txt
2022-09-28 03:02:59 -05:00
Vivek Khandelwal
28daf410b6
Add instructions to use locally build Torch-MLIR with SHARK
2022-09-28 10:16:38 +05:30
Ean Garvey
cbf3f784aa
Add pytest option to specify a URL for shark tank artifacts. ( #363 )
...
* Xfail updates.
* Generalize tank SHA option to bucket address and add pytest option.
2022-09-27 02:40:40 -05:00
Anush Elangovan
ef4b306c7b
Add diffusers and scipy
2022-09-26 13:35:23 -07:00
powderluv
5316c1e0bf
Use latest transformers ( #346 )
2022-09-26 13:11:41 -07:00
Gaurav Shukla
0228973eef
[WEB] Fix the mlir location of stable-diffusion model ( #367 )
...
Update the location of stable-diffusion mlir file since there is some
problem with iree-compile.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2022-09-26 10:56:36 -07:00
Gaurav Shukla
d4eeff0a5d
[WEB] Add Stable-Diffusion in the SHARK web ( #366 )
...
1. This commit adds stable-diffusion as a part of shark web.
2. The V-diffusion model has been disabled for now as it's not
working(will raise a different patch with fix).
3. Add standard output in the web ui.
4. Add instructions to launch the shark-web.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2022-09-26 10:42:02 -07:00
Prashant Kumar
c7b2d39ab2
Update stable_diff to contain vae.
2022-09-26 20:11:43 +05:30
Gaurav Shukla
21958cc02a
[WEB] Remove unused parameters in the v-diffuison model ( #314 )
...
This commit removes unused parameters in the v-diffusion model. It also
updated the server parameters in order to make multiple requests to be
handled sequentially.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2022-09-25 10:57:06 -07:00
Ean Garvey
de23e5d9d7
update xfails for PyTorch DistilBERT ( #355 )
2022-09-24 14:53:20 -05:00
Quinn Dawkins
6438bce023
Add a script to convert a jpg to the correct input for resnet50 with the vulkan gui ( #362 )
2022-09-23 16:32:52 -07:00
yzhang93
587d74b449
Update model annotation tool ( #361 )
...
Usage:
with create_context() as ctx:
module = model_annotation(ctx, input_contents=..., config_path=..., search_op=...)
Example:
The example is to annotate the minilm model with GPU config files.
python model_annotation.py /nodclouddata/vivian/minilm_model/model.mlir /nodclouddata/vivian/minilm_model/model_config.json
2022-09-23 15:44:51 -07:00
Prashant Kumar
b9c8985047
Add sharkdynamo which combines shark with torchdynamo.
...
-- Adds graph breaks when necessary.
-- Even for loops are supported.
2022-09-23 22:40:02 +05:30
Vivek Khandelwal
93ebe07d2b
Add bert_tosa script
2022-09-23 10:52:06 +05:30
Ean Garvey
d82b305781
Fix issues with loading .vmfb into SharkInference
2022-09-23 09:53:13 +05:30
Quinn Dawkins
1df20fac95
[Lockstep] Hack to avoid aten._reshape_alias ( #332 )
...
This enforces the decomposition for aten._reshape_alias used in AOTAutograd to essentially avoid having to deal with problems with strides when running in eager mode.
2022-09-22 18:02:09 -04:00
Prashant Kumar
991e7043d1
Add stable diffusion model.
2022-09-22 13:40:51 +05:30
powderluv
1c4d6c23fa
Update CMakeLists.txt
2022-09-21 22:48:56 -07:00
Anush Elangovan
87895446a5
Roll SHARK-Runtime
2022-09-22 00:09:04 -07:00
Ean Garvey
c0f3a09a40
Include SHA in path to failure reproducers. Add --save_fails option. ( #352 )
2022-09-21 17:55:06 -05:00
Anush Elangovan
e9ad4b9fc4
Update SHARK Runtime
2022-09-21 06:31:48 -07:00
Ean Garvey
c061a8897d
Add pytest options to save reproducers. ( #350 )
...
* Add pytest options to save and/or upload reproducers.
* pass shark_module to benchmark method.
2022-09-20 20:29:46 -05:00
Ean Garvey
4253551b67
Update README with new testing instructions and filter test cases. ( #349 )
2022-09-20 15:55:46 -05:00
Vivek Khandelwal
e4991c049e
Add Readme file for the bloom model
2022-09-20 20:27:52 +05:30
Daniel Garvey
5df582e7e8
creates abstract test case class ( #333 )
2022-09-20 07:06:38 -07:00
Ean Garvey
814a6f8295
Modify vulkan target triple substring searches. ( #318 )
2022-09-20 01:20:20 -05:00
Vivek Khandelwal
7013c3cd4a
Add bloom e2e script
2022-09-20 10:56:04 +05:30
powderluv
0ddd65b6f1
Create LICENSE
2022-09-19 15:07:59 -07:00
powderluv
44d8f08bfc
Fix Torch-MLIR release page
2022-09-17 00:50:39 -07:00
erman-gurses
fc8aa6ae63
Add ROCM parameters ( #335 )
2022-09-16 09:12:19 -07:00
Quinn Dawkins
9bd951b083
Clean up the v-diffusion install pipeline ( #327 )
2022-09-16 11:47:07 -04:00
Vivek Khandelwal
c43448a826
Update compile_utils.py
2022-09-15 18:28:10 +05:30
Vivek Khandelwal
864723a473
add bloom model example
2022-09-15 18:23:09 +05:30
Anush Elangovan
3b0ec8ce4e
Update resnet paths
2022-09-14 16:56:20 -07:00
Anush Elangovan
174b171913
Clean up SDL linking
2022-09-14 13:18:55 -07:00
powderluv
cfd9733c2b
Delete shark_web directory
2022-09-14 06:38:30 -07:00
Anush Elangovan
8d4d543a49
Update shark runtime
2022-09-14 06:14:02 -07:00
Anush Elangovan
1b9c88a052
Update vulkan gui readme
2022-09-13 19:35:47 -07:00
Anush Elangovan
e212ff2071
Fix resnet50 vulkan_gui to work with tank models
2022-09-13 19:22:41 -07:00