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
Quinn Dawkins
8d21292d34
Fix input tensors with non-floating point dtype in the lockstep tracer ( #328 )
2022-09-13 21:14:38 -04: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
Anush Elangovan
23378b6be8
Add resnet to vulkan-gui
2022-09-13 07:06:47 -07:00
Ean Garvey
6cf5564c84
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. ( #325 )
...
* Replace instances of "gpu" alias for devices with "cuda"
2022-09-13 01:16:56 -05:00
Ean Garvey
7143902a90
Update test-models.yml ( #323 )
2022-09-12 22:47:40 -05:00
Anush Elangovan
15186db73f
Hardcode SDL2 for now (works on linux)
2022-09-12 10:17:41 -07:00
powderluv
ccd7a01ce2
Update README.md
2022-09-12 07:12:57 -07:00
Anush Elangovan
1d7035117d
Add cpp inference examples and vulkan_gui
2022-09-12 07:07:33 -07:00
Ean Garvey
1710abd366
Update mobilenet_v3_small_torch_test.py ( #322 )
2022-09-10 15:22:57 -05:00
Ean Garvey
6aeda3670f
Split nightly workflow by backend (IREE / SHARK) ( #313 )
...
* Fix validation for nightly builds.
* Add option to generate shark_tank inside SHARK project
Add shark_arg for updating tank on mismatched hash (downloader)
* Fixup CI tank dir option.
* Fixup work directory variable
2022-09-09 22:51:30 +05:30
Prashant Kumar
bb52b224d0
Add sparse architecture and test with torchrec SparseArch.
...
Features that don't work with current implementation:
-- embeddingbag config with multiple features.
2022-09-09 21:49:30 +05:30
Stanley Winata
95ec3d7216
[tank][v-diffusion] Polish up v-diffusion UX ( #315 )
2022-09-08 12:55:51 -07:00
powderluv
18872222d3
Update README.md
2022-09-07 01:14:30 -07:00
Ean Garvey
d453f2e49d
Enable CPU benchmarks on test-models workflows. ( #299 )
...
* Update test-models.yml
* Update README.md
2022-09-07 01:22:58 -05:00
Ean Garvey
3824d37d27
Add metadata to benchmark results. ( #297 )
2022-09-06 13:03:48 -05:00
Ean Garvey
d946287723
Update xfails for torchvision models. ( #310 )
2022-09-01 13:06:12 -05:00
Prashant Kumar
885b0969f5
[WEB] Cache the compiled module.
...
-- Don't compile the module again and again.
2022-09-01 23:08:08 +05:30
Gaurav Shukla
a886cba655
[WEB] Add v_diffusion model in the shark web ( #306 )
...
This commit adds adds `v_diffusion` model web visualization as a part of
shark web.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2022-09-01 06:34:51 -07:00
Vivek Khandelwal
4afe2e3adb
Add func to save intermediate images in v_diffusion_pytorch
2022-09-01 18:36:58 +05:30
Gaurav Shukla
fe080eaee6
[WEB] Introduce web interface for the SHARK models ( #298 )
...
This commit introduces web application for SHARK using gradio platform.
This adds web visualization of `Resnet50` and `Albert_Maskfill` models
as a start.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2022-08-31 23:17:52 -07:00
Quinn Dawkins
3703f014d9
Add scripts for generating images on ats-m ( #305 )
2022-08-31 23:07:02 -07:00
Daniel Garvey
d45a496030
adds a flag to enable directory choice ( #303 )
...
individual tests will require implementation of the flag
alternatively, simply passing shark_default_sha in your
individual app's download function will allow for this behavior
2022-08-31 22:17:40 -07:00
powderluv
4ee164c66f
remove a100 cpu
2022-08-31 12:59:47 -07:00
powderluv
bf84c033bb
add icelake
2022-08-31 12:58:40 -07:00
Prashant Kumar
5105f62551
Add the dlrm_model in shark example. ( #301 )
...
-- DLRM model is added in the shark example.
-- The model is verified on cpu, gpu and vulkan.
Co-authored-by: powderluv <powderluv@users.noreply.github.com >
2022-08-31 12:54:21 -07:00
Quinn Dawkins
99be837d84
Add lockstep tracer based on TorchMLIR eager mode + examples ( #243 )
2022-08-31 15:50:24 -04:00
Quinn Dawkins
b7766898ee
Add cfg sampling from tank model for v-diffusion and move compilation outside of the sampling loop ( #302 )
2022-08-31 11:35:04 -07:00
powderluv
57f73dfbc9
Update nightly.yml
2022-08-28 23:59:03 -07:00
powderluv
50b2b9638d
Update nightly.yml
2022-08-28 23:43:32 -07:00
Daniel Garvey
1bfd00e2f8
fixes an install issue ( #295 )
2022-08-25 18:52:00 -05: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
Phaneesh Barwaria
02d857260c
Update ReadMe
...
-Add gsutil installation for resnet50 example
2022-08-25 20:28:50 +05:30
Phaneesh Barwaria
1322ec5935
Simplified Testing Interface ( #289 )
2022-08-24 23:54:56 -05:00
Daniel Garvey
48e9818f7e
adds support to default to tuned model ( #287 )
...
currently setup for tf bert/resnet50
going to refactor test class to avoid having to
add an argument to 50+ files
2022-08-24 16:30:02 -05:00
Ean Garvey
14857770dc
Fix local artifact recognition and usage by SHARK downloader. ( #286 )
...
* Fix local artifact recognition and usage by SHARK downloader.
* Update generate_sharktank.py
* Update generate_sharktank.py
2022-08-24 14:37:16 -05:00
Vivek Khandelwal
f79a6bf5aa
Update setup_v_diffusion_pytorch.sh ( #291 )
...
Fix minor issue with v-diffusion PyTorch version
2022-08-24 22:00:02 +05:30
Prashant Kumar
7dc27a7477
Don't remove the latest .whl package from CI. ( #290 )
...
Previously, the CI was removing the latest package and pointing to the
stale package.
2022-08-24 09:03:48 -07:00
Chi_Liu
17dba601c8
Add huggingface top5 image classification automodel ( #268 )
2022-08-22 15:05:38 -07:00
Chi_Liu
064aa3b1f4
Fix tmp dir bug ( #285 )
2022-08-22 15:00:35 -07:00
Ean Garvey
4960efc686
Update requirements-importer.txt ( #284 )
2022-08-19 23:21:41 -05:00