|
|
@@ -22,6 +22,7 @@ |
|
|
|
#include "common/blocking_queue.h" |
|
|
|
#include "common/properties_manager.h" |
|
|
|
#include "framework/common/debug/ge_log.h" |
|
|
|
#include "graph/ge_local_context.h" |
|
|
|
#include "hybrid/common/npu_memory_allocator.h" |
|
|
|
#include "hybrid/common/tensor_value.h" |
|
|
|
#include "hybrid/executor/hybrid_profiler.h" |
|
|
@@ -38,6 +39,7 @@ struct GraphExecutionContext { |
|
|
|
|
|
|
|
uint64_t session_id = 0; |
|
|
|
const HybridModel *model = nullptr; |
|
|
|
const GEThreadLocalContext *ge_context = nullptr; |
|
|
|
rtStream_t stream = nullptr; |
|
|
|
rtContext_t rt_context = nullptr; |
|
|
|
rtContext_t rt_gen_context = nullptr; |
|
|
|