Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

support the dynamic sequence in the heterogeneous

This commit is contained in:
limingqi107 2023-02-02 21:00:56 +08:00
parent 05fc4fc280
commit d6bc1b3d06
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_actor.cc
View File

@ -479,7 +479,7 @@ void KernelActor::CopyInputDeviceTensor(const OpData<DeviceTensor> *input_data,
auto &new_device_tensor = copy_input_device_tensors_[input_data_index];
MS_EXCEPTION_IF_NULL(new_device_tensor);
// Dynamic shape need update size.
if (IsDynamic(real_input_info->shape_)) {
if (IsDynamic(real_input_info->shape_) || common::AnfAlgo::IsDynamicSequence(kernel_)) {
new_device_tensor->SetSize(input_data->data_->GetSize());
}
// Update the input device tensor.