Cose belle

2026-07-14 16:57:58 +02:00
parent d1a29ace3c
commit 51fdb830e5
38 changed files with 5365 additions and 914 deletions
@@ -8,6 +8,7 @@
 #include "llvm/Support/LogicalResult.h"

 #include "src/Accelerators/PIM/Common/PimCommon.hpp"
+#include "src/Accelerators/PIM/Common/IR/ConstantUtils.hpp"
 #include "src/Accelerators/PIM/Dialect/Spatial/SpatialOps.hpp"

 using namespace mlir;
@@ -130,7 +131,7 @@ struct CanonicalizeSingleLaneComputeBatchPattern : OpRewritePattern<ComputeBatch
    rewriter.setInsertionPointToStart(newBlock);

    IRMapping mapper;
-    Value zero = arith::ConstantIndexOp::create(rewriter, compute.getLoc(), 0);
+    Value zero = getOrCreateIndexConstant(rewriter, compute.getOperation(), 0);
    mapper.map(*oldLaneArg, zero);
    for (auto [index, weight] : llvm::enumerate(compute.getWeights())) {
      auto oldArg = compute.getWeightArgument(index);