From 6aebb84e143c024ea7a805e85149177f89c5cf38 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Zientkiewicz?= Date: Thu, 8 Aug 2024 20:07:18 +0200 Subject: [PATCH] Add workarounds for the temporary absence of exec2.h. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Michał Zientkiewicz --- .../executor/executor2/stream_assignment.h | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/dali/pipeline/executor/executor2/stream_assignment.h b/dali/pipeline/executor/executor2/stream_assignment.h index 6e8b25fbea..0f1d69b415 100644 --- a/dali/pipeline/executor/executor2/stream_assignment.h +++ b/dali/pipeline/executor/executor2/stream_assignment.h @@ -26,11 +26,24 @@ #include #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 class StreamAssignment;