Use converted type to generate constant.

lmendesp-amd · lmendesp-amd · commit 918d71942232 · 2024-12-09T13:43:30.000Z
diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp
@@ -61,7 +61,7 @@ static Value createLinalgBodyCalculationForElementwiseOp(
 
   if (isa<tosa::AbsOp>(op) && isa<IntegerType>(elementTy)) {
     auto zero = rewriter.create<arith::ConstantOp>(
-        loc, rewriter.getZeroAttr(elementTy));
+        loc, rewriter.getZeroAttr(convertedElementTy));
     auto neg = rewriter.create<arith::SubIOp>(loc, zero, args[0]);
     return rewriter.create<arith::MaxSIOp>(loc, args[0], neg);
   }

Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@ static Value createLinalgBodyCalculationForElementwiseOp(`
`61`	`61`
`62`	`62`	`if (isa<tosa::AbsOp>(op) && isa<IntegerType>(elementTy)) {`
`63`	`63`	`auto zero = rewriter.create<arith::ConstantOp>(`
`64`		`- loc, rewriter.getZeroAttr(elementTy));`
	`64`	`+ loc, rewriter.getZeroAttr(convertedElementTy));`
`65`	`65`	`auto neg = rewriter.create<arith::SubIOp>(loc, zero, args[0]);`
`66`	`66`	`return rewriter.create<arith::MaxSIOp>(loc, args[0], neg);`
`67`	`67`	`}`