Skip to content

Commit a567a54

Browse files
add comment
1 parent 6c6ed79 commit a567a54

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -810,6 +810,9 @@ static Value calculateGatherOffset(RewriterBase &rewriter,
810810

811811
enum VectorMemoryAccessKind { ScalarBroadcast, Contiguous, Gather };
812812

813+
/// Find the non constant dim in a linalgOp. This is used for finding contiguous
814+
/// loads and it is expected that only one dim will be non constant, if thats
815+
/// not the case this function will assert.
813816
static uint64_t getNonUnitLoopDim(LinalgOp linalgOp) {
814817
uint64_t nonUnitDim = 0;
815818
uint64_t countNonUnitDim = 0;

0 commit comments

Comments
 (0)