[mlir][spirv] Add GPU subgroup MMA to spirv.MMAMatrixTimesScalar
authorLei Zhang <antiagainst@google.com>
Mon, 5 Dec 2022 22:18:34 +0000 (22:18 +0000)
committerLei Zhang <antiagainst@google.com>
Mon, 5 Dec 2022 22:30:50 +0000 (22:30 +0000)
commit2c7827da4f5bf758a1659a0c4a2d0e7944827c42
treee6e7d6571174723da30b56e7373b4cd250fe7a42
parent96d63993dd3698bbf2d6a83c035cd05faed7317b
[mlir][spirv] Add GPU subgroup MMA to spirv.MMAMatrixTimesScalar

Along the way, make the default pattern fail instead of crashing
when an elementwise op is not supported yet.

Reviewed By: kuhar

Differential Revision: https://reviews.llvm.org/D139280
mlir/lib/Conversion/GPUToSPIRV/WmmaOpsToSPIRV.cpp
mlir/test/Conversion/GPUToSPIRV/wmma-ops-to-spirv.mlir