Browse Source

fix

pull/277/head
wangxiaotian22 4 years ago
parent
commit
7f076840f9
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      ge/graph/passes/hccl_memcpy_pass.cc
  2. +1
    -1
      ge/graph/passes/hccl_memcpy_pass.h

+ 2
- 2
ge/graph/passes/hccl_memcpy_pass.cc View File

@@ -160,12 +160,12 @@ Status HcclMemcpyPass::P2pmemInputProcess(const ComputeGraphPtr &graph, const No
return SUCCESS; return SUCCESS;
} }


for (int index = 0; index < input_memory_types.size() && index < op_desc->GetInputsSize(); index++) {
for (uint32_t index = 0; index < input_memory_types.size() && index < op_desc->GetInputsSize(); index++) {
if (input_memory_types[index] != RT_MEMORY_P2P_DDR) { if (input_memory_types[index] != RT_MEMORY_P2P_DDR) {
continue; continue;
} }


auto hccl_in_anchor = GetInDataAnchor(index);
auto hccl_in_anchor = node->GetInDataAnchor(index);
if (hccl_in_anchor == nullptr) { if (hccl_in_anchor == nullptr) {
continue; continue;
} }


+ 1
- 1
ge/graph/passes/hccl_memcpy_pass.h View File

@@ -43,7 +43,7 @@ class HcclMemcpyPass : public GraphPass {


Status P2pmemInputProcess(const ComputeGraphPtr &graph, const NodePtr node); Status P2pmemInputProcess(const ComputeGraphPtr &graph, const NodePtr node);


bool HcclMemcpyPass::IsDataNode(const std::string& node_type);
bool IsDataNode(const std::string& node_type);


std::unordered_map<std::string, uint32_t> node_num_map_; std::unordered_map<std::string, uint32_t> node_num_map_;
}; };


Loading…
Cancel
Save