AMD

Target-specific Gluon APIs for AMD GPU generations.

Common APIs

AMDMFMALayout

Represents a layout for AMD MFMA (matrix core) operations.

AMDWMMALayout

Represents a layout for AMD WMMA (matrix core) operations.

slice

Extract a register only slice of a tensor.

warp_pipeline_stage

Marks a warp-pipeline stage inside a Gluon kernel.

GPU Generations