Browse Source

skip parameter output of kernel graph.

feature/build-system-rewrite
gaoyong10 4 years ago
parent
commit
10d5e45b36
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/runtime/graph_scheduler/control_node_parser.cc

+ 1
- 0
mindspore/ccsrc/runtime/graph_scheduler/control_node_parser.cc View File

@@ -1843,6 +1843,7 @@ void CollectEffectiveOutputByGraph(const KernelGraphPtr &graph, DeviceContext *c

for (const auto &backend_to_front : graph->graph_output_map()) {
if (HasAbstractMonad(backend_to_front.second.first) || HasAbstractMonad(backend_to_front.first.first) ||
backend_to_front.first.first->isa<Parameter>() ||
common::AnfAlgo::CheckPrimitiveType(backend_to_front.second.first, prim::kPrimPartial) ||
backend_to_front.second.first->isa<ValueNode>()) {
continue;


Loading…
Cancel
Save