Skip to content

Commit 6aebb84

Browse files
committed
Add workarounds for the temporary absence of exec2.h.
Signed-off-by: Michał Zientkiewicz <[email protected]>
1 parent 538eb3b commit 6aebb84

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

dali/pipeline/executor/executor2/stream_assignment.h

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,24 @@
2626
#include <vector>
2727
#include "dali/pipeline/graph/graph_util.h"
2828
#include "dali/pipeline/executor/executor2/exec_graph.h"
29-
#include "dali/pipeline/executor/executor2/exec2.h"
29+
// TODO(michalz): This is here for review process only. Remove when exec2.h is available
30+
// #include "dali/pipeline/executor/executor2/exec2.h"
31+
#include "dali/pipeline/graph/op_graph2.h"
3032

3133
namespace dali {
3234
namespace exec2 {
3335

36+
// TODO(michalz): This is here for review process only. Remove when exec2.h is available
37+
enum class StreamPolicy : int {
38+
Single, //< There's just one stream that's used by all operators
39+
PerBackend, //< Operators are scheduled on a stream specific to their backend (mixed or GPU)
40+
PerOperator //< Independent operators are executed on separate streams.
41+
42+
// TODO(michalz): Check if this is legal with existing operator implementations - likely not
43+
// PerIteration, //< Streams are cycled on a per-iteration basis
44+
};
45+
46+
3447
template <StreamPolicy policy>
3548
class StreamAssignment;
3649

0 commit comments

Comments
 (0)