[Mlir-commits] [mlir] [mlir][amdgpu] Define an amdgpu.scaling_mfma wrapper (PR #137498)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Sun Apr 27 22:48:46 PDT 2025
================
@@ -833,6 +833,14 @@ mfmaOpToScaledIntrinsic(MFMAOp mfma, Chipset chipset) {
mfma.getBlocks(), chipset);
}
+static std::optional<std::tuple<StringRef, uint32_t, uint32_t>>
+mfmaOpToScaledIntrinsic(ScaledMFMAOp smfma, Chipset chipset) {
+ return mfmaOpToScaledIntrinsic(smfma.getSourceA().getType(),
----------------
Muzammiluddin-Syed-ECE wrote:
IIUC, templating doesn't remove the need for branching since I would still be branching off the class of the input op to pass in the appropriate arguments. So, I opted for the branching on op type which I find clearer.
https://github.com/llvm/llvm-project/pull/137498
More information about the Mlir-commits
mailing list