[Mlir-commits] [mlir] Extend NVVM mma ops to support fp64 (PR #165380)
Durgadoss R
llvmlistbot at llvm.org
Fri Oct 31 04:50:03 PDT 2025
================
@@ -954,6 +960,14 @@ LogicalResult NVVM::WMMALoadOp::verify() {
return emitOpError() << "invalid attribute combination";
std::pair<Type, unsigned> typeInfo = inferMMATypeFromMNK(
getEltype(), getFrag(), getM(), getN(), getK(), getContext());
+ // Special case for f64 fragments
+ Type f64Ty = Float64Type::get(getContext());
+ if (typeInfo.first == f64Ty && typeInfo.second == 1) {
+ if (getType() != f64Ty)
+ return emitOpError("expected destination type to be f64");
----------------
durga4github wrote:
Thank you!
https://github.com/llvm/llvm-project/pull/165380
More information about the Mlir-commits
mailing list