github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-10 06:17:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add gpu tested models.

Browse Source
This commit is contained in:
Prashant Kumar
2022-05-24 05:55:24 +00:00
parent 4b63e0e04a
commit 126771fa55
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@@ -70,17 +70,17 @@ result = shark_module.forward(inputs)
| MobileNetV2 | :heavy_check_mark: (Script) | | | |
| MobileNetV3 | :heavy_check_mark: (Script) | | | |
| Unet | :x: (Script) | | | |
| Resnet18 | :heavy_check_mark: (Script) | :heavy_check_mark: | | |
| Resnet50 | :heavy_check_mark: (Script) | :heavy_check_mark: | | |
| Resnet18 | :heavy_check_mark: (Script) | :heavy_check_mark: | :heavy_check_mark: | |
| Resnet50 | :heavy_check_mark: (Script) | :heavy_check_mark: | :heavy_check_mark: | |
| Resnext50_32x4d | :heavy_check_mark: (Script) | | | |
| ShuffleNet_v2 | :x: (Script) | | | |
| SqueezeNet | :heavy_check_mark: (Script) | :heavy_check_mark: | | |
| SqueezeNet | :heavy_check_mark: (Script) | :heavy_check_mark: | :heavy_check_mark: | |
| EfficientNet | :heavy_check_mark: (Script) | | | |
| Regnet | :heavy_check_mark: (Script) | | | |
| Resnest | :x: (Script) | | | |
| Vision Transformer | :heavy_check_mark: (Script) | | | |
| VGG 16 | :heavy_check_mark: (Script) | | | |
| Wide Resnet | :heavy_check_mark: (Script) | :heavy_check_mark: | | |
| Wide Resnet | :heavy_check_mark: (Script) | :heavy_check_mark: | :heavy_check_mark: | |
| RAFT | :x: (JIT) | | | |
For more information refer to [MODEL TRACKING SHEET](https://docs.google.com/spreadsheets/d/15PcjKeHZIrB5LfDyuw7DGEEE8XnQEX2aX8lm8qbxV8A/edit#gid=0)

2
shark/iree_utils.py
View File

@@ -73,7 +73,7 @@ def get_iree_compiled_module(module, device: str):
args.append(f"-iree-llvm-target-triple={target_triple}")
if device == "gpu":
args += ["--iree-cuda-llvm-target-arch=sm_80", "--iree-hal-cuda-disable-loop-nounroll-wa"]
args += ["--iree-hal-cuda-disable-loop-nounroll-wa"]
ireert.flags.FUNCTION_INPUT_VALIDATION = False
ireert.flags.parse_flags("--cuda_allow_inline_execution")