llvm-project
b414d779 - [MLIR][NVVM] Unify and move to a single tcgen05_mma_kind attr for all tcgen05.mma Ops (#184433)

Commit
23 days ago
[MLIR][NVVM] Unify and move to a single tcgen05_mma_kind attr for all tcgen05.mma Ops (#184433) This change unifies using of `tcgen05_mma_kind` attribute for tcgen05.mma Ops in MLIR. Before this change there were two block scale attributes used for tcgen05.mma Ops. One was `MMABlockScaleKindAttr` with `mxf8f6f4`, `mxf4` and `fxf4nvf4` values used for `tcgen05.mma.block_scale` and `tcgen05.mma.sp.block_scale`. Another one was `Tcgen05MMAKindAttr` with `f16`, `tf32`, `f8f6f4` and `i8` values used for `tcgen05.mma`, `tcgen05.mma.sp`, `tcgen05.mma.ws` and `tcgen05.mma.ws.sp`. `Tcgen05MMAKindAttr` has been extended with values from `MMABlockScaleKindAttr`. Now there is `tcgen05_mma_kind` attribute only for all `tcgen05.mma` Ops in MLIR. Backward compatibility is not supported. Existing tests and scripts should be updated to use `tcgen05_mma_kind` attribute instead of `block_scale_kind` for all tcgen05.mma MLIR Ops.
Author
Parents
Loading