[MLIR][NVVM] Introduce WGMMA Types
This work introduces `WGMMATypes` attributes for the `WgmmaMmaSyncOp`. This op, having been recently added to MLIR, previously used `MMATypes`. However, there arises a disparity in supported types between `MmaOp` and `WgmmaMmaSyncOp`. To address this discrepancy more effectively, a new set of attributes is introduced. Furthermore, this patch refines and optimizing the verification mechanisms of `WgmmaMmaSyncOp` Op. It also adds support for f8 types, including `e4m3` and `e5m2`, within the `WgmmaMmaSyncOp`. Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D157695
Loading
Please sign in to comment