uniquify constants produced by affine lowering

2026-06-01 10:52:25 +02:00
parent b678e55d3c
commit 356be6ccc2
3 changed files with 70 additions and 0 deletions
@@ -26,6 +26,7 @@
 #include <utility>

 #include "Common/PimCommon.hpp"
+#include "Common/IR/ConstantUtils.hpp"
 #include "Conversion/ONNXToSpatial/Common/Common.hpp"
 #include "Conversion/SpatialToPim/Common.hpp"
 #include "Conversion/SpatialToPim/Patterns.hpp"
@@ -264,6 +265,8 @@ void onnx_mlir::raptor::SpatialToPimPass::runOnOperation() {
    signalPassFailure();
    return;
  }
+  hoistAndUniquifyIndexConstants(funcOp, rewriter);
+
  // Dump to file for debug
  dumpModule(moduleOp, "pim0");
 }