- Implements GPU WMMA ops to SPIRV Cooperative matrix conversion
- Relaxes the memory requirements on Memref's used in GPU MMA ops as SPIRV has different address mappings.
- Allows SPIRV unary arithmetic ops to also take cooperative matrix type as the SPIRV Cooperative matrix spec allows for negate ops which is unary.
- Extends the elementwise op types supported by SubgroupMmaElementwiseOp to allow the ops supported by the SPIRV Cooperative matrix spec.
Details
Details
Diff Detail
Diff Detail
Event Timeline
Comment Actions
LGTM, just one style related comment
mlir/lib/Conversion/GPUToSPIRV/WmmaOpsToSPIRV.cpp | ||
---|---|---|
116 | Spell out type here and below when the type is not obvious: |
nit: fix formatting