Skip to content

Commit 1b08c0a

Browse files
committed
[mlir][vector] Use getShapeForUnroll's default implementation.
1 parent fb21f16 commit 1b08c0a

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def Vector_MultiDimReductionOp :
360360

361361
def Vector_BroadcastOp :
362362
Vector_Op<"broadcast", [Pure,
363-
DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
363+
DeclareOpInterfaceMethods<VectorUnrollOpInterface>,
364364
DeclareOpInterfaceMethods<InferIntRangeInterface, ["inferResultRanges"]>,
365365
PredOpTrait<"source operand and result have same element type",
366366
TCresVTEtIsSameAsOpBase<0, 0>>]>,

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

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2782,10 +2782,6 @@ void BroadcastOp::inferResultRanges(ArrayRef<ConstantIntRanges> argRanges,
27822782
setResultRanges(getResult(), argRanges.front());
27832783
}
27842784

2785-
std::optional<SmallVector<int64_t, 4>> BroadcastOp::getShapeForUnroll() {
2786-
return llvm::to_vector<4>(getResultVectorType().getShape());
2787-
}
2788-
27892785
/// Return the dimensions of the result vector that were formerly ones in the
27902786
/// source tensor and thus correspond to "dim-1" broadcasting.
27912787
static llvm::SetVector<int64_t>

0 commit comments

Comments
 (0)