We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent adde9a9 commit d8b806aCopy full SHA for d8b806a
mlir/lib/Dialect/Linalg/Transforms/MeshShardingInterfaceImpl.cpp
@@ -47,6 +47,7 @@ using MeshShardingAttr = mesh::MeshShardingAttr;
47
using ShardingArray = mesh::ShardingArray;
48
using MeshOp = mesh::MeshOp;
49
50
+// Returns the corresponding mesh reduction kind for the given arith op.
51
static ReductionKind getReductionKind(Operation *op) {
52
return llvm::TypeSwitch<Operation *, ReductionKind>(op)
53
// Floating-point operations.
0 commit comments