| @@ -36,6 +36,7 @@ namespace { | |||||
| const int kAllInputAddrIsAtomic = -1; | const int kAllInputAddrIsAtomic = -1; | ||||
| const int kVirtualInputNodeMemoryReuse = 0; | const int kVirtualInputNodeMemoryReuse = 0; | ||||
| const int kVirtualOutputNodeMemoryReuse = 1; | const int kVirtualOutputNodeMemoryReuse = 1; | ||||
| const int kPrevNextDistanceNum = 2; | |||||
| const int64_t kInvalidStream = -1; | const int64_t kInvalidStream = -1; | ||||
| const char *const kEngineNameGeLocal = "DNN_VM_GE_LOCAL_OP_STORE"; | const char *const kEngineNameGeLocal = "DNN_VM_GE_LOCAL_OP_STORE"; | ||||
| // One state per bit cannot be repeated | // One state per bit cannot be repeated | ||||
| @@ -1979,7 +1980,7 @@ static void UpdatePrevNodeInputDesc(const NodePtr &prev_node, | |||||
| continue; | continue; | ||||
| } | } | ||||
| if (prev_next_distances.size() == 2) { | |||||
| if (prev_next_distances.size() == kPrevNextDistanceNum) { | |||||
| prev_next_distances[1] = distance; | prev_next_distances[1] = distance; | ||||
| } else { | } else { | ||||
| GELOGW("Size of prev_next_distances is not 2."); | GELOGW("Size of prev_next_distances is not 2."); | ||||
| @@ -2114,7 +2115,7 @@ void GraphMemoryAssigner::MarkNodeDistanceAttr(const ComputeGraphPtr &compute_gr | |||||
| GELOGW("Get ATTR_NAME_DATA_VISIT_DISTANCE failed."); | GELOGW("Get ATTR_NAME_DATA_VISIT_DISTANCE failed."); | ||||
| continue; | continue; | ||||
| } | } | ||||
| if (prev_next_distances.size() != 2) { | |||||
| if (prev_next_distances.size() != kPrevNextDistanceNum) { | |||||
| GELOGW("Size of prev_next_distance is not 2."); | GELOGW("Size of prev_next_distance is not 2."); | ||||
| continue; | continue; | ||||
| } else { | } else { | ||||