5.5 KiB
5.5 KiB
Compatibility Matrix
The tables below show mutually exclusive features and the support on some hardware.
The symbols used have the following meanings:
- ✅ = Full compatibility
- 🟠 = Partial compatibility
- ❌ = No compatibility
- ❔ = Unknown or TBD
!!! note Check the ❌ or 🟠 with links to see tracking issue for unsupported feature/hardware combination.
Feature x Feature
<style> td:not(:first-child) { text-align: center !important; } td { padding: 0.5rem !important; white-space: nowrap; } th { padding: 0.5rem !important; min-width: 0 !important; } th:not(:first-child) { writing-mode: vertical-lr; transform: rotate(180deg) } </style>Feature | [CP][chunked-prefill] | APC | LoRA | SD | CUDA graph | pooling | enc-dec | logP | prmpt logP | async output | multi-step | mm | best-of | beam-search |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
[CP][chunked-prefill] | ✅ | |||||||||||||
APC | ✅ | ✅ | ||||||||||||
LoRA | ✅ | ✅ | ✅ | |||||||||||
SD | ✅ | ✅ | ❌ | ✅ | ||||||||||
CUDA graph | ✅ | ✅ | ✅ | ✅ | ✅ | |||||||||
pooling | ✅* | ✅* | ✅ | ❌ | ✅ | ✅ | ||||||||
enc-dec | ❌ | ❌ | ❌ | ❌ | ✅ | ✅ | ✅ | |||||||
logP | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ✅ | ||||||
prmpt logP | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ | |||||
async output | ✅ | ✅ | ✅ | ❌ | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ||||
multi-step | ❌ | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ✅ | |||
mm | ✅ | ✅ | 🟠 | ❔ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❔ | ✅ | ||
best-of | ✅ | ✅ | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ✅ | ❔ | ❌ | ✅ | ✅ | |
beam-search | ✅ | ✅ | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ✅ | ❔ | ❌ | ❔ | ✅ | ✅ |
* Chunked prefill and prefix caching are only applicable to last-token pooling.
Feature x Hardware
Feature | Volta | Turing | Ampere | Ada | Hopper | CPU | AMD | TPU |
---|---|---|---|---|---|---|---|---|
[CP][chunked-prefill] | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
APC | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
LoRA | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
SD | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
CUDA graph | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ❌ |
pooling | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
enc-dec | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | ❌ |
mm | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
logP | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
prmpt logP | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
async output | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ |
multi-step | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ❌ |
best-of | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
beam-search | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
!!! note Please refer to [Feature support through NxD Inference backend][feature-support-through-nxd-inference-backend] for features supported on AWS Neuron hardware