fix much stuff

2026-05-22 18:53:38 +02:00
parent 8337a11ce9
commit 2c1da813b5
18 changed files with 502 additions and 191 deletions
@@ -1,3 +1,5 @@
+#include <cassert>
+
 #include "src/Accelerators/PIM/Conversion/SpatialToPim/ComputeLikeRegionUtils.hpp"
 #include "src/Accelerators/PIM/Dialect/Spatial/SpatialOps.hpp"

@@ -29,9 +31,17 @@ void replaceAndEraseDirectComputeLikeInput(PatternRewriter& rewriter,
                                           unsigned inputIndex,
                                           Value replacement) {
  Block& body = owner->getRegion(0).front();
-  BlockArgument bodyArgument = isa<spatial::SpatCompute>(owner)
-                               ? cast<spatial::SpatCompute>(owner).getInputArgument(inputIndex)
-                               : cast<spatial::SpatComputeBatch>(owner).getInputArgument(inputIndex);
+  BlockArgument bodyArgument;
+  if (auto compute = dyn_cast<spatial::SpatCompute>(owner)) {
+    auto computeArg = compute.getInputArgument(inputIndex);
+    assert(computeArg && "expected compute input block argument");
+    bodyArgument = *computeArg;
+  }
+  else {
+    auto batchArg = cast<spatial::SpatComputeBatch>(owner).getInputArgument(inputIndex);
+    assert(batchArg && "expected compute_batch input block argument");
+    bodyArgument = *batchArg;
+  }
  unsigned bodyArgIndex = bodyArgument.getArgNumber();

  rewriter.startOpModification(owner);