Skip to content

Commit

Permalink
Add workarounds for the temporary absence of exec2.h.
Browse files Browse the repository at this point in the history
Signed-off-by: Michał Zientkiewicz <[email protected]>
  • Loading branch information
mzient committed Aug 8, 2024
1 parent 6e8426d commit 4b35a70
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion dali/pipeline/executor/executor2/stream_assignment.h
Original file line number Diff line number Diff line change
Expand Up @@ -26,11 +26,24 @@
#include <vector>
#include "dali/pipeline/graph/graph_util.h"
#include "dali/pipeline/executor/executor2/exec_graph.h"
#include "dali/pipeline/executor/executor2/exec2.h"
// TODO(michalz): This is here for review process only. Remove when exec2.h is available
// #include "dali/pipeline/executor/executor2/exec2.h"
#include "dali/pipeline/graph/op_graph2.h"

namespace dali {
namespace exec2 {

// TODO(michalz): This is here for review process only. Remove when exec2.h is available
enum class StreamPolicy : int {
Single, //< There's just one stream that's used by all operators
PerBackend, //< Operators are scheduled on a stream specific to their backend (mixed or GPU)
PerOperator //< Independent operators are executed on separate streams.

// TODO(michalz): Check if this is legal with existing operator implementations - likely not
// PerIteration, //< Streams are cycled on a per-iteration basis
};


template <StreamPolicy policy>
class StreamAssignment;

Expand Down

0 comments on commit 4b35a70

Please sign in to comment.