Skip to content

Commit dbf24a8

Browse files
author
Liang-Ta Wei
committed
create tosa.constOp instead
1 parent 5f55822 commit dbf24a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/lib/Dialect/Tosa/IR/TosaCanonicalizations.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ struct ConsolidateTransposeOptimization
370370
RankedTensorType::get(transposePerms.size(), rewriter.getI32Type());
371371
auto permsAttr = DenseIntElementsAttr::get(permsTy, perms);
372372
Value permsValue =
373-
rewriter.create<arith::ConstantOp>(transposeOp.getLoc(), permsAttr);
373+
rewriter.create<tosa::ConstOp>(transposeOp.getLoc(), permsTy, permsAttr);
374374

375375
rewriter.replaceOpWithNewOp<tosa::TransposeOp>(
376376
transposeOp, transposeOp.getResult().getType(),

0 commit comments

Comments
 (0)