File tree Expand file tree Collapse file tree 1 file changed +14
-1
lines changed
dali/pipeline/executor/executor2 Expand file tree Collapse file tree 1 file changed +14
-1
lines changed Original file line number Diff line number Diff line change 26
26
#include < vector>
27
27
#include " dali/pipeline/graph/graph_util.h"
28
28
#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"
30
32
31
33
namespace dali {
32
34
namespace exec2 {
33
35
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
+
34
47
template <StreamPolicy policy>
35
48
class StreamAssignment ;
36
49
You can’t perform that action at this time.
0 commit comments