Skip to content

Commit fbbf0e4

Browse files
committed
[mlir][vector] Use getShapeForUnroll's default implementation.
1 parent 4a7d3df commit fbbf0e4

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
@@ -361,7 +361,7 @@ def Vector_MultiDimReductionOp :
361361

362362
def Vector_BroadcastOp :
363363
Vector_Op<"broadcast", [Pure,
364-
DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
364+
DeclareOpInterfaceMethods<VectorUnrollOpInterface>,
365365
DeclareOpInterfaceMethods<InferIntRangeInterface, ["inferResultRanges"]>,
366366
PredOpTrait<"source operand and result have same element type",
367367
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)