[Mlir-commits] [mlir] [mlir][mesh] Add collective communication operations (PR #71960)
Chengji Yao
llvmlistbot at llvm.org
Tue Nov 14 14:29:18 PST 2023
================
@@ -471,46 +460,42 @@ LogicalResult verifyReduceScatterOperandAndResultShape(
//===----------------------------------------------------------------------===//
LogicalResult
-mlir::mesh::AllReduceOp::verifySymbolUses(SymbolTableCollection &symbolTable) {
+AllReduceOp::verifySymbolUses(SymbolTableCollection &symbolTable) {
return verifyMeshSymbolUses(*this, symbolTable);
}
-void mlir::mesh::AllReduceOp::getCanonicalizationPatterns(
- RewritePatternSet &patterns, MLIRContext *context) {
- populateMeshAxesSetCanonicalizationPatterns<AllReduceOp>(patterns, context);
+LogicalResult mlir::mesh::AllReduceOp::verify() {
+ return verifyMeshAxes(getLoc(), getMeshAxes());
----------------
yaochengji wrote:
Do you mean this check?
```
if (!meshAxes) {
return success();
}
```
But an ArrayAttr with an empty array is not nullptr.
https://github.com/llvm/llvm-project/pull/71960
More information about the Mlir-commits
mailing list