Skip to content

Commit d8b806a

Browse files
committed
Add a comment for getReductionKind
1 parent adde9a9 commit d8b806a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ using MeshShardingAttr = mesh::MeshShardingAttr;
4747
using ShardingArray = mesh::ShardingArray;
4848
using MeshOp = mesh::MeshOp;
4949

50+
// Returns the corresponding mesh reduction kind for the given arith op.
5051
static ReductionKind getReductionKind(Operation *op) {
5152
return llvm::TypeSwitch<Operation *, ReductionKind>(op)
5253
// Floating-point operations.

0 commit comments

Comments
 (0)