Reset all outputs before running an operator.

Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
NVIDIA · Sep 6, 2022 · db026f0 · db026f0
1 parent 41e8107
commit db026f0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dali/pipeline/executor/executor.cc b/dali/pipeline/executor/executor.cc
@@ -327,9 +327,13 @@ void Executor<WorkspacePolicy, QueuePolicy>::RunHelper(OpNode &op_node, Workspac
 
   for (int i = 0; i < ws.NumOutput(); i++) {
     if (ws.template OutputIsType<CPUBackend>(i)) {
-      set_order(ws.template Output<CPUBackend>(i));
+      auto &out = ws.template Output<CPUBackend>(i);
+      out.Reset();
+      set_order(out);
     } else {
-      set_order(ws.template Output<GPUBackend>(i));
+      auto &out = ws.template Output<GPUBackend>(i);
+      out.Reset();
+      set_order(out);
     }
   }