Remove getShapeForUnroll for step

Author: nbpatel

mlir/include/mlir/Dialect/Vector/IR/VectorOps.td

@@ -3018,7 +3018,7 @@ def Vector_ScanOp :
 
 def Vector_StepOp : Vector_Op<"step", [
     Pure,
-    DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
+    DeclareOpInterfaceMethods<VectorUnrollOpInterface>,
     DeclareOpInterfaceMethods<InferIntRangeInterface, ["inferResultRanges"]>
   ]> {
   let summary = "A linear sequence of values from 0 to N";

mlir/lib/Dialect/Vector/IR/VectorOps.cpp

@@ -7442,10 +7442,6 @@ void StepOp::inferResultRanges(ArrayRef<ConstantIntRanges> argRanges,
   setResultRanges(getResult(), result);
 }
 
-std::optional<SmallVector<int64_t, 4>> StepOp::getShapeForUnroll() {
-  return llvm::to_vector<4>(llvm::cast<VectorType>(getType()).getShape());
-}
-
 //===----------------------------------------------------------------------===//
 // Vector Masking Utilities
 //===----------------------------------------------------------------------===//