Skip to content

Commit ec0e502

Browse files
committed
*Updated code comment related to broadcast map check.
1 parent 8c329f4 commit ec0e502

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3458,7 +3458,10 @@ static LogicalResult verifyExtendedMatmulSemantic(MatmulOp matmulOp,
34583458
return matmulOp->emitOpError()
34593459
<< "Unexpected dim expression in map result.";
34603460

3461-
// Check if the requested broadcast is valid.
3461+
// Check if the user defined map is valid broadcast map. Here broadcast
3462+
// indexing maps are defined in context of corresponding default indexing maps
3463+
// for the given Op. This way the check becomes very simple i.e just check the
3464+
// number of result dims.
34623465
if (isBroadcasted(opIndexingMap, defaultIndexingMap)) {
34633466
if (!matmulOp.isValidLhsRhsBroadcastMap(opIndexingMap)) {
34643467
return matmulOp->emitOpError()

0 commit comments

Comments
 (0)