Browse Source

fix compile error

tags/v1.2.0
wangxiaotian22 3 years ago
parent
commit
ce22b66201
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      ge/graph/manager/graph_manager.cc
  2. +1
    -1
      ge/graph/manager/graph_manager.h

+ 2
- 1
ge/graph/manager/graph_manager.cc View File

@@ -2558,7 +2558,8 @@ Status GraphManager::RunGraphAsync(const GraphId &graph_id, const std::vector<ge
uint64_t session_id, RunAsyncCallback callback) {
GELOGI("[GraphManager] Start to run graph async, graph_id=%u, inputsSize=%zu.", graph_id, inputs.size());

bool ret = prerun_args_q_.Push(PreRunArgs({graph_id, inputs, session_id, GetThreadLocalContext(), callback}));
bool ret = prerun_args_q_.Push(PreRunArgs({graph_id, inputs, session_id,
GetContext().WorkStreamId(), GetThreadLocalContext(), callback}));
if (!ret) {
GELOGE(FAILED, "[GraphManager] Run graph async failed, graph_id=%u.", graph_id);
return FAILED;


+ 1
- 1
ge/graph/manager/graph_manager.h View File

@@ -223,7 +223,7 @@ class GraphManager {
const SubGraphInfoPtr &sub_graph_info_ptr,
const std::string &root_graph_name,
uint64_t session_id,
uint64_t work_stream_id;
uint64_t work_stream_id,
const GEThreadLocalContext &ge_context);
Status ParseInputsDims(const std::vector<InputTensorInfo> &input_tensor);
void ParseInputsDimsForData(const std::vector<InputTensorInfo> &input_tensor);


Loading…
Cancel
Save