lichun
4 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
+ 2
- 2
ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc
View File
| @@ -22,6 +22,7 @@ | |||||
| #include "graph/utils/node_utils.h" | #include "graph/utils/node_utils.h" | ||||
| #include "graph/utils/tensor_utils.h" | #include "graph/utils/tensor_utils.h" | ||||
| #include "graph/utils/type_utils.h" | #include "graph/utils/type_utils.h" | ||||
| #include "graph/compute_graph.h" | |||||
| #include "ge_local_engine/ops_kernel_store/op/op_factory.h" | #include "ge_local_engine/ops_kernel_store/op/op_factory.h" | ||||
| #include "ge_local_engine/common/constant/constant.h" | #include "ge_local_engine/common/constant/constant.h" | ||||
| #include "register/ops_kernel_builder_registry.h" | #include "register/ops_kernel_builder_registry.h" | ||||
| @@ -164,8 +165,7 @@ Status GeLocalOpsKernelBuilder::CalcConstantStrMemSize(const OpDescPtr &op_desc, | |||||
| Status GeLocalOpsKernelBuilder::GenerateTask(const Node &node, RunContext &context, std::vector<domi::TaskDef> &tasks) { | Status GeLocalOpsKernelBuilder::GenerateTask(const Node &node, RunContext &context, std::vector<domi::TaskDef> &tasks) { | ||||
| bool is_in_unknown_subgraph = false; | bool is_in_unknown_subgraph = false; | ||||
| bool forced_unknown = false; | bool forced_unknown = false; | ||||
| ComputeGraphPtr owner_graph = node.GetOwnerComputeGraph(); | |||||
| for (const auto &node : owner_graph->GetDirectNode()) { | |||||
| for (const auto &node : node.GetOwnerComputeGraph()->GetDirectNode()) { | |||||
| GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_subgraph), | GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_subgraph), | ||||
| "[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str()); | "[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str()); | ||||
| if (is_in_unknown_subgraph) { | if (is_in_unknown_subgraph) { | ||||