handle getNextOutput and contiguous conflict

pull/8548/head
laiyongqiang 4 years ago
parent 22b6ff6214
commit 0fdc6b547d

@ -69,12 +69,12 @@ bool Somas::Allocate(const session::KernelGraph *graph) {
bool Somas::InitSomasTensors(const session::KernelGraph *graph) { bool Somas::InitSomasTensors(const session::KernelGraph *graph) {
MS_EXCEPTION_IF_NULL(graph); MS_EXCEPTION_IF_NULL(graph);
InitBasicInfo(graph); InitBasicInfo(graph);
GetNextOutputProcess(graph);
IndependentNodeOutputProcess(graph); IndependentNodeOutputProcess(graph);
SummaryInputProcess(graph); SummaryInputProcess(graph);
RefNodeProcess(graph); RefNodeProcess(graph);
UnReuseNodeProcess(graph); UnReuseNodeProcess(graph);
GenContiguousList(graph); GenContiguousList(graph);
GetNextOutputProcess(graph);
if (tensors_list_.empty()) { if (tensors_list_.empty()) {
MS_LOG(INFO) << "No Tensor from graph " << graph->graph_id(); MS_LOG(INFO) << "No Tensor from graph " << graph->graph_id();
@ -273,6 +273,7 @@ void Somas::GetNextOutputProcess(const session::KernelGraph *graph) {
if (iter != nodes_map_.end()) { if (iter != nodes_map_.end()) {
auto getnext_output_tensors = iter->second->output_tensors_; auto getnext_output_tensors = iter->second->output_tensors_;
for (auto &tensor : getnext_output_tensors) { for (auto &tensor : getnext_output_tensors) {
if (tensor->contiguous_) continue;
tensor->offset_ = total_size; tensor->offset_ = total_size;
total_size += tensor->GetAlignedSize(); total_size += tensor->GetAlignedSize();
tensor->lifelong_value_ = kLifeLongGraphAll; tensor->lifelong_value_ = kLifeLongGraphAll;

Loading…
Cancel
Save