add

ita9naiwa · ita9naiwa · commit be6a11935796 · 2025-02-28T16:19:20.000+09:00
diff --git a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
@@ -5001,9 +5001,12 @@ static bool isLikePadUnPad(PackOrUnpackOp packOp, ShapedType packedTensorType) {
 }
 
 bool PackOp::isLikePad() {
-  auto packedTensorType =
-      llvm::cast<RankedTensorType>((*this)->getResultTypes().front());
-  return isLikePadUnPad(*this, packedTensorType);
+  if (auto packedTensorType =
+          llvm::dyn_cast<RankedTensorType>((*this)->getResultTypes().front()))
+    return isLikePadUnPad(*this, packedTensorType);
+  if (auto packedTensorType =
+          llvm::dyn_cast<MemRefType>((*this)->getResultTypes().front()))
+    return isLikePadUnPad(*this, packedTensorType);
 }
 
 OpFoldResult PackOp::fold(FoldAdaptor adaptor) {

Original file line number	Diff line number	Diff line change
`@@ -5001,9 +5001,12 @@ static bool isLikePadUnPad(PackOrUnpackOp packOp, ShapedType packedTensorType) {`
`5001`	`5001`	`}`
`5002`	`5002`
`5003`	`5003`	`bool PackOp::isLikePad() {`
`5004`		`- auto packedTensorType =`
`5005`		`- llvm::cast<RankedTensorType>((*this)->getResultTypes().front());`
`5006`		`- return isLikePadUnPad(*this, packedTensorType);`
	`5004`	`+ if (auto packedTensorType =`
	`5005`	`+ llvm::dyn_cast<RankedTensorType>((*this)->getResultTypes().front()))`
	`5006`	`+ return isLikePadUnPad(*this, packedTensorType);`
	`5007`	`+ if (auto packedTensorType =`
	`5008`	`+ llvm::dyn_cast<MemRefType>((*this)->getResultTypes().front()))`
	`5009`	`+ return isLikePadUnPad(*this, packedTensorType);`
`5007`	`5010`	`}`
`5008`	`5011`
`5009`	`5012`	`OpFoldResult PackOp::fold(FoldAdaptor adaptor) {`