[llvm] [mlir] [mlir][mesh] Add all-scatter operation (PR #81218)

Lei Zhang via llvm-commits llvm-commits at lists.llvm.org
Tue Feb 13 12:01:59 PST 2024


================
@@ -9,17 +9,34 @@
 #ifndef MLIR_DIALECT_MESH_TRANSFORMS_TRANSFORMS_H
 #define MLIR_DIALECT_MESH_TRANSFORMS_TRANSFORMS_H
 
+#include "mlir/Dialect/Mesh/IR/MeshOps.h"
+#include "mlir/IR/BuiltinTypes.h"
+#include "mlir/IR/Value.h"
+#include "mlir/Support/LLVM.h"
+
 namespace mlir {
 class RewritePatternSet;
 class SymbolTableCollection;
 class DialectRegistry;
+class ImplicitLocOpBuilder;
 namespace mesh {
 
 void processMultiIndexOpLoweringPopulatePatterns(
     RewritePatternSet &patterns, SymbolTableCollection &symbolTableCollection);
-
 void processMultiIndexOpLoweringRegisterDialects(DialectRegistry &registry);
 
+void allScatterOpLoweringPopulatePatterns(
----------------
antiagainst wrote:

`populateAllScatterOpLoweringPatterns` to follow the baming convention?

https://github.com/llvm/llvm-project/pull/81218


More information about the llvm-commits mailing list