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. |
|
Create partitioned shared memory layouts and WMMA layout for a CDNA5 GEMM in order to avoid LDS partition conflicts. |
|
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. |