Rebase and adjust.

Signed-off-by: Michał Zientkiewicz <[email protected]>
NVIDIA · Aug 9, 2024 · 258ff3b · 258ff3b
1 parent c507643
commit 258ff3b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dali/pipeline/executor/executor2/exec_graph_test.cc b/dali/pipeline/executor/executor2/exec_graph_test.cc
@@ -466,11 +466,11 @@ TEST(ExecGraphTest, LoweredExec) {
   env.thread_pool = &tp;
   params.env = &env;
   params.batch_size = 32;
-  auto iter = std::make_shared<IterationData>();
+  params.iter_data = std::make_shared<IterationData>();
   {
     tasking::Executor ex(4);
     ex.Start();
-    g.PrepareIteration(iter, params);
+    g.PrepareIteration(params);
     auto fut = g.Launch(ex);
     auto &out = fut.Value<const PipelineOutput &>();
     CheckTestGraph1Results(out.workspace, params.batch_size);