Skip to content

Commit 2b5d1dc

Browse files
committed
Amending comments of getConvOperationKind()
1 parent 13f5183 commit 2b5d1dc

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1948,7 +1948,10 @@ static bool isCastOfBlockArgument(Operation *op) {
19481948
isa<BlockArgument>(op->getOperand(0));
19491949
}
19501950

1951-
// Returns true iff it is a valid conv/pooling op.
1951+
// Returns the ConvOperationKind of the op using reduceOp of the generic
1952+
// payload. If it is neither a convolution nor a pooling, it returns
1953+
// std::nullopt.
1954+
//
19521955
// If (region has 2 ops (reduction + yield) or 3 ops (extension + reduction
19531956
// + yield) and rhs is not used) then it is the body of a pooling
19541957
// If conv, check for single `mul` predecessor. The `mul` operands must be

0 commit comments

Comments
 (0)