Fix memory bug

2026-06-03 12:59:58 +02:00
parent 0a5e73c3ea
commit 3c2667f11e
1 changed files with 1 additions and 1 deletions
@@ -140,7 +140,7 @@ void PimMemory::allocateHost(ModuleOp moduleOp, func::FuncOp funcOp) {
  });

  funcOp.walk([&](memref::AllocOp allocOp) {
-    if (!allocOp->getParentOfType<pim::PimCoreOp>())
+    if (!allocOp->getParentOfType<pim::PimCoreOp>() && !allocOp->getParentOfType<pim::PimCoreBatchOp>())
      gatherMemEntry(allocOp.getResult());
  });