|
|
|
@ -470,6 +470,7 @@ void Somas::GenContiguousList(const session::KernelGraph *graph) {
|
|
|
|
|
if (node->GetType() != kCommunicationNode) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
if ((!node->input_tensors_.empty()) && (!node->input_tensors_[0]->contiguous_)) {
|
|
|
|
|
std::vector<size_t> inputs;
|
|
|
|
|
auto input_before_gap = CreateGapTensor(gap_tensor_id);
|
|
|
|
|
input_before_gap->contiguous_ = true;
|
|
|
|
@ -487,7 +488,9 @@ void Somas::GenContiguousList(const session::KernelGraph *graph) {
|
|
|
|
|
input_after_gap->contiguous_ = true;
|
|
|
|
|
inputs.push_back(input_after_gap->GetId());
|
|
|
|
|
contiguous_tensors_list_.push_back(inputs);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if ((!node->output_tensors_.empty()) && (!node->output_tensors_[0]->contiguous_)) {
|
|
|
|
|
std::vector<size_t> outputs;
|
|
|
|
|
auto output_before_gap = CreateGapTensor(gap_tensor_id);
|
|
|
|
|
gap_tensor_id++;
|
|
|
|
@ -506,6 +509,7 @@ void Somas::GenContiguousList(const session::KernelGraph *graph) {
|
|
|
|
|
outputs.push_back(output_after_gap->GetId());
|
|
|
|
|
contiguous_tensors_list_.push_back(outputs);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Somas::PreprocessingConflicts() {
|
|
|
|
|