Fixup a bug caused by dealloc event's async dependency list (Xilinx#472)

nod-ai · Mar 5, 2024 · 0340ada · 0340ada
1 parent 5243a62
commit 0340ada
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlir/lib/Transform/AIRDependencyScheduleOpt.cpp b/mlir/lib/Transform/AIRDependencyScheduleOpt.cpp
@@ -3694,6 +3694,7 @@ class AIRSegmentLoopFusion
       air::ExecuteOp dealloc_exec = execOpPair.second;
       dealloc_exec->moveBefore(new_loop_op.getBody(),
                                new_loop_op.getBody()->getOperations().end());
+      clearAsyncDependenciesOfAsyncOp(dealloc_exec);
     }
 
     // Scf.yield op.