github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md with model tracking information

Browse Source 
Hugging face models and TorchVision models tracking sheet has been added.
This commit is contained in:
Prashant Kumar
2022-03-30 19:33:38 +05:30
committed by GitHub
parent 1cdc9cd198
commit f30d24ab2b
1 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
shark_runner/README.md
View File

@@ -37,3 +37,38 @@ result = shark_module.forward(inputs)
### Shark Trainer API
#### Work in Progress
### Model Tracking (Shark Inference)
| Hugging Face Models | Torch-MLIR lowerable | IREE-CPU | IREE-GPU | IREE-VULKAN |
|---------------------|----------------------|----------|----------|-------------|
| BERT | Yes (JIT) | | | |
| Albert | Yes (JIT) | | | |
| BigBird | Yes (AOT) | | | |
| DistilBERT | Yes (AOT) | | | |
| GPT2 | Yes (AOT) | | | |
| TORCHVISION Models | Torch-MLIR lowerable | IREE-CPU | IREE-GPU | IREE-VULKAN |
|--------------------|----------------------|----------|----------|-------------|
| AlexNet | Yes (Script) | | | |
| DenseNet121 | Yes (Script) | | | |
| MNasNet1_0 | Yes (Script) | | | |
| MobileNetV2 | Yes (Script) | | | |
| MobileNetV3 | Yes (Script) | | | |
| Unet | Yes (Script) | | | |
| Resnet18 | Yes (Script) | | | |
| Resnet50 | Yes (Script) | | | |
| Resnext50_32x4d | Yes (Script) | | | |
| ShuffleNet_v2 | Yes (Script) | | | |
| SqueezeNet | Yes (Script) | | | |
| EfficientNet | Yes (Script) | | | |
| Regnet | Yes (Script) | | | |
| Resnest | Yes (Script) | | | |
| Vision Transformer | Yes (Script) | | | |
| VGG 16 | Yes (Script) | | | |
| Wide Resnet | Yes (Script) | | | |
| RAFT | Yes (JIT) | | | |
For more information refer to [MODEL TRACKING SHEET](https://docs.google.com/spreadsheets/d/15PcjKeHZIrB5LfDyuw7DGEEE8XnQEX2aX8lm8qbxV8A/edit#gid=0)