AMD CDNA 5
AMD buffer load from global memory via a scalar base pointer and a tensor of offsets instead of a tensor of pointers. |
|
AMD buffer store a tensor directly to global memory via a scalar base pointer and a tensor of offsets instead of a tensor of pointers. |
|
Get the scale layout for WMMA scaled operands. |
|
Load M/N-packed fp4 bytes from shared memory into a K-packed WMMA dot operand layout. |
|
Create partitioned shared memory layouts and WMMA layout for a CDNA5 GEMM in order to avoid LDS partition conflicts. |
|
Scale and convert FP16, BF16, or FP32 values to a low-precision MX format, dividing by the raw E8M0 |
|
Upcast an fp4 or fp8 tensor and fold raw E8M0 scale payload into the CDNA5 scaled-upcast op. |
|
Computes matrix multiplication |
|
AMD Scaled WMMA operation. |
|
Represents a partitioned shared memory layout that splits a tensor across multiple physical shared memory partitions. |