[mlir][vector] Use getShapeForUnroll's default implementation.

amd-eochoalo · amd-eochoalo · commit 5103187a4f7b · 2025-11-06T15:06:12.000-05:00
diff --git a/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td b/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
@@ -1653,7 +1653,7 @@ def Vector_TransferWriteOp :
 }
 
 def Vector_LoadOp : Vector_Op<"load", [
-    DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
+    DeclareOpInterfaceMethods<VectorUnrollOpInterface>,
     DeclareOpInterfaceMethods<MemorySpaceCastConsumerOpInterface>,
     DeclareOpInterfaceMethods<AlignmentAttrOpInterface>
   ]> {
diff --git a/mlir/lib/Dialect/Vector/IR/VectorOps.cpp b/mlir/lib/Dialect/Vector/IR/VectorOps.cpp
@@ -5762,10 +5762,6 @@ OpFoldResult LoadOp::fold(FoldAdaptor) {
   return OpFoldResult();
 }
 
-std::optional<SmallVector<int64_t, 4>> LoadOp::getShapeForUnroll() {
-  return llvm::to_vector<4>(getVectorType().getShape());
-}
-
 FailureOr<std::optional<SmallVector<Value>>>
 LoadOp::bubbleDownCasts(OpBuilder &builder) {
   return mlir::detail::bubbleDownInPlaceMemorySpaceCastImpl(getBaseMutable(),

Original file line number	Diff line number	Diff line change
`@@ -1653,7 +1653,7 @@ def Vector_TransferWriteOp :`
`1653`	`1653`	`}`
`1654`	`1654`
`1655`	`1655`	`def Vector_LoadOp : Vector_Op<"load", [`
`1656`		`- DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,`
	`1656`	`+ DeclareOpInterfaceMethods<VectorUnrollOpInterface>,`
`1657`	`1657`	`DeclareOpInterfaceMethods<MemorySpaceCastConsumerOpInterface>,`
`1658`	`1658`	`DeclareOpInterfaceMethods<AlignmentAttrOpInterface>`
`1659`	`1659`	`]> {`