Browse Source

!12032 Build ops pynative bprop graph on ascend

From: @HulkTang
Reviewed-by: @chujinjin,@kisnwang
Signed-off-by: @chujinjin
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
08ffd9dcbe
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/backend/session/ascend_session.cc

+ 2
- 2
mindspore/ccsrc/backend/session/ascend_session.cc View File

@@ -701,7 +701,7 @@ void AscendSession::GetOpInputStubTensors(const CNodePtr &cnode, const std::map<

void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfNodePtr, size_t> &parameter_index,
const std::vector<tensor::TensorPtr> &graph_inputs) {
if (built_graph_id_.find(graph_id) == built_graph_id_.end()) {
if (built_graph_id_.find(graph_id) != built_graph_id_.end()) {
return;
}
auto graph = GetGraph(graph_id);
@@ -751,7 +751,7 @@ void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfN
for (const auto &single_op_graph : single_op_graphs) {
for (const auto &graph_info : single_op_graph.second) {
run_op_graphs_[graph_info] = single_op_graph.first;
MS_LOG(DEBUG) << "Pre build op finished, graph info: " << single_op_graph.second;
MS_LOG(DEBUG) << "Pre build op finished, graph info: " << graph_info;
}
}
built_graph_id_.insert(graph_id);


Loading…
Cancel
Save