Skip to content

Featuresยถ

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ยถ

Feature CP APC LoRA SD CUDA graph pooling enc-dec logP prmpt logP async output multi-step mm best-of beam-search prompt-embeds
CP โœ…
APC โœ… โœ…
LoRA โœ… โœ… โœ…
SD โœ… โœ… โŒ โœ…
CUDA graph โœ… โœ… โœ… โœ… โœ…
pooling ๐ŸŸ * ๐ŸŸ * โœ… โŒ โœ… โœ…
enc-dec โŒ โŒ โŒ โŒ โœ… โœ… โœ…
logP โœ… โœ… โœ… โœ… โœ… โŒ โœ… โœ…
prmpt logP โœ… โœ… โœ… โœ… โœ… โŒ โœ… โœ… โœ…
async output โœ… โœ… โœ… โŒ โœ… โŒ โŒ โœ… โœ… โœ…
multi-step โŒ โœ… โŒ โŒ โœ… โŒ โŒ โœ… โœ… โœ… โœ…
mm โœ… โœ… ๐ŸŸ ^ โ” โœ… โœ… โœ… โœ… โœ… โœ… โ” โœ…
best-of โœ… โœ… โœ… โŒ โœ… โŒ โœ… โœ… โœ… โ” โŒ โœ… โœ…
beam-search โœ… โœ… โœ… โŒ โœ… โŒ โœ… โœ… โœ… โ” โŒ โ” โœ… โœ…
prompt-embeds โœ… โœ… โœ… โŒ โœ… โŒ โŒ โœ… โŒ โ” โ” โœ… โ” โ” โœ…

* Chunked prefill and prefix caching are only applicable to last-token or all pooling with causal attention.
^ LoRA is only applicable to the language backbone of multimodal models.

Feature x Hardwareยถ

Feature Volta Turing Ampere Ada Hopper CPU AMD Intel GPU
CP โŒ โœ… โœ… โœ… โœ… โœ… โœ… โœ…
APC โŒ โœ… โœ… โœ… โœ… โœ… โœ… โœ…
LoRA โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…
SD โœ… โœ… โœ… โœ… โœ… โŒ โœ… โœ…
CUDA graph โœ… โœ… โœ… โœ… โœ… โŒ โœ… โŒ
pooling โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…
enc-dec โœ… โœ… โœ… โœ… โœ… โœ… โŒ โœ…
mm โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…
prompt-embeds โœ… โœ… โœ… โœ… โœ… โœ… โ” โœ…
logP โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…
prmpt logP โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…
async output โœ… โœ… โœ… โœ… โœ… โŒ โŒ โœ…
multi-step โœ… โœ… โœ… โœ… โœ… โŒ โœ… โœ…
best-of โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…
beam-search โœ… โœ… โœ… โœ… โœ… โœ… โœ… โœ…

Note

For information on feature support on Google TPU, please refer to the TPU-Inference Recommended Models and Features documentation.