Expand description
Intel Xe Matrix Extensions (XMX) SPIR-V kernel generators.
XMX is Intel’s matrix-multiply-accumulate hardware (analogous to NVIDIA
Tensor Cores or AMD MFMA). It is exposed via the
SPV_KHR_cooperative_matrix SPIR-V extension and the
CooperativeMatrixKHR capability (SPIR-V 1.6 / GLSL 4.6).
This module provides:
XmxTileConfig— tile dimension configuration for XMX GEMM.gemm_xmx_spirv— cooperative-matrix GEMM kernel (C = alpha*A*B + beta*C) targeting Intel Xe / Arc / Ponte Vecchio with XMX engines.gemm_xmx_f16_spirv— FP16 input / FP32 accumulation variant.matmul_xmx_bf16_spirv— BF16 input / FP32 accumulation variant.
Structs§
- XmxTile
Config - Tile dimensions for XMX GEMM.
Functions§
- best_
xmx_ tile - Best XMX tile configuration for the given device name.
- device_
supports_ xmx - Whether the given Level Zero device name suggests XMX hardware support.
- gemm_
xmx_ f16_ spirv - Generate a SPIR-V binary for an XMX-accelerated FP16→FP32 GEMM kernel.
- gemm_
xmx_ spirv - Generate a SPIR-V binary for an XMX-accelerated FP32 GEMM kernel.
- matmul_
xmx_ bf16_ spirv - Generate a SPIR-V binary for BF16 input / FP32 accumulation XMX GEMM.