Skip to content

Commit

Permalink
Executor2 class implementation & tests (#5528)
Browse files Browse the repository at this point in the history
* Add Executor2 class, implementing ExecutorBase interface and using exec2::ExecGraph
* Use a queue of futures to represent the output buffer queue.
* Add CI test jobs
* Add an environment variable that allows the user to use executor 2.0 instead of the default AsyncPipelinedExecutor

---------

Signed-off-by: Michal Zientkiewicz <[email protected]>
  • Loading branch information
mzient authored Sep 10, 2024
1 parent 2117f88 commit f1a9a4d
Show file tree
Hide file tree
Showing 17 changed files with 931 additions and 33 deletions.
6 changes: 3 additions & 3 deletions dali/c_api/c_api_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -622,9 +622,9 @@ TYPED_TEST(CApiTest, TestExecutorMeta) {

pipe_ptr.reset();
daliPipelineHandle handle;
daliCreatePipeline(&handle, serialized.c_str(), serialized.size(), batch_size, num_thread,
this->device_id_, false, prefetch_queue_depth, prefetch_queue_depth,
prefetch_queue_depth, true);
daliCreatePipeline2(&handle, serialized.c_str(), serialized.size(), batch_size, num_thread,
this->device_id_, false, false, false,
prefetch_queue_depth, prefetch_queue_depth, prefetch_queue_depth, true);

daliRun(&handle);
daliOutput(&handle);
Expand Down
Loading

0 comments on commit f1a9a4d

Please sign in to comment.