Transpose and Refactor of Patterns

2026-05-29 13:23:31 +02:00
parent 82b44a6387
commit 832bd7f1f7
37 changed files with 285 additions and 153 deletions
@@ -0,0 +1,40 @@
+#include "src/Accelerators/PIM/Conversion/SpatialToPim/Patterns.hpp"
+#include "src/Accelerators/PIM/Conversion/SpatialToPim/Common.hpp"
+#include "src/Accelerators/PIM/Dialect/Pim/PimOps.hpp"
+
+using namespace mlir;
+
+namespace onnx_mlir {
+namespace raptor {
+
+#include "src/Accelerators/PIM/Conversion/SpatialToPim/SpatialToPim.hpp.inc"
+
+} // namespace raptor
+
+void populateInitialPatterns(RewritePatternSet& patterns) {
+  raptor::populateWithGenerated(patterns);
+  populateTransposeLoweringPatterns(patterns);
+}
+
+void populateGlobalTensorMaterializationPatternPhase(RewritePatternSet& patterns) {
+  populateGlobalTensorMaterializationPatterns(patterns);
+}
+
+void populateInitialTensorPackingPatterns(RewritePatternSet& patterns) {
+  populateTensorPackingPatterns(patterns);
+}
+
+void populateCoreBodyPatterns(RewritePatternSet& patterns) {
+  raptor::populateWithGenerated(patterns);
+  populateTransposeLoweringPatterns(patterns);
+}
+
+void populateFinalTensorPackingPatterns(RewritePatternSet& patterns) {
+  populateTensorPackingPatterns(patterns);
+}
+
+void populateCommunicationPatterns(RewritePatternSet& patterns) {
+  populateChannelLoweringPatterns(patterns);
+}
+
+} // namespace onnx_mlir