Build variants

A kernel can be compliant for a specific compute framework (e.g. CUDA) or architecture (e.g. x86_64). For compliance with a compute framework and architecture combination, all the build variants listed below must be available. This list will be updated as new PyTorch versions are released.

CPU aarch64-darwin

• torch213-cpu-aarch64-darwin
• torch214-cpu-aarch64-darwin

Metal aarch64-darwin

• torch210-metal-aarch64-darwin
• torch211-metal-aarch64-darwin
• torch212-metal-aarch64-darwin
• torch213-metal-aarch64-darwin
• torch214-metal-aarch64-darwin

CPU aarch64-linux

• torch213-cpu-aarch64-linux
• torch214-cpu-aarch64-linux

CUDA aarch64-linux

• torch213-cu126-aarch64-linux
• torch213-cu130-aarch64-linux
• torch213-cu132-aarch64-linux
• torch214-cu126-aarch64-linux
• torch214-cu130-aarch64-linux
• torch214-cu132-aarch64-linux

CPU x86_64-linux

• torch213-cpu-x86_64-linux
• torch214-cpu-x86_64-linux

CUDA x86_64-linux

• torch213-cu126-x86_64-linux
• torch213-cu130-x86_64-linux
• torch213-cu132-x86_64-linux
• torch214-cu126-x86_64-linux
• torch214-cu130-x86_64-linux
• torch214-cu132-x86_64-linux

ROCm x86_64-linux

• torch213-rocm71-x86_64-linux
• torch213-rocm72-x86_64-linux
• torch214-rocm714-x86_64-linux
• torch214-rocm72-x86_64-linux

XPU x86_64-linux

• torch210-xpu20253-x86_64-linux
• torch211-xpu20253-x86_64-linux
• torch212-xpu20253-x86_64-linux
• torch213-xpu20260-x86_64-linux
• torch214-xpu20261-x86_64-linux

Python-only kernels

Kernels that are in pure Python (e.g. Triton kernels) only need to provide one or more of the following variants:

• torch-cpu
• torch-cuda
• torch-metal
• torch-rocm
• torch-xpu