We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
tosa.pad
1 parent ea85900 commit 106e367Copy full SHA for 106e367
mlir/lib/Dialect/Tosa/IR/TosaCanonicalizations.cpp
@@ -878,8 +878,9 @@ OpFoldResult ReshapeOp::fold(FoldAdaptor adaptor) {
878
OpFoldResult PadOp::fold(FoldAdaptor adaptor) {
879
// If the pad is all zeros we can fold this operation away.
880
if (adaptor.getPadding() && getInput1().getType() == getType()) {
881
- auto densePad = llvm::cast<DenseElementsAttr>(adaptor.getPadding());
882
- if (densePad.isSplat() && densePad.getSplatValue<APInt>().isZero()) {
+ auto densePad = llvm::dyn_cast<DenseElementsAttr>(adaptor.getPadding());
+ if (densePad && densePad.isSplat() &&
883
+ densePad.getSplatValue<APInt>().isZero()) {
884
return getInput1();
885
}
886
0 commit comments