diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp index 8373808e0fc6334..22c506ba3031521 100644 --- a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp +++ b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp @@ -61,7 +61,7 @@ static Value createLinalgBodyCalculationForElementwiseOp( if (isa(op) && isa(elementTy)) { auto zero = rewriter.create( - loc, rewriter.getZeroAttr(elementTy)); + loc, rewriter.getZeroAttr(convertedElementTy)); auto neg = rewriter.create(loc, zero, args[0]); return rewriter.create(loc, args[0], neg); }