|
|
|
@@ -931,9 +931,8 @@ bool KernelRuntime::LaunchKernelMod(const session::KernelGraph &graph) { |
|
|
|
<< " should be equal to the size of kernels " << kernels.size(); |
|
|
|
} |
|
|
|
for (size_t i = 0; i < kernels.size(); ++i) { |
|
|
|
auto &kernel = kernels[i]; |
|
|
|
if (!dynamic_kernel_list.empty() && dynamic_kernel_list[i] != nullptr && |
|
|
|
dynamic_kernel_list[i]->is_dynamic_shape() && AnfAlgo::GetKernelType(kernel) == AICPU_KERNEL) { |
|
|
|
dynamic_kernel_list[i]->is_dynamic_shape()) { |
|
|
|
dynamic_kernel_list[i]->InferShape(); |
|
|
|
dynamic_kernel_list[i]->UpdateArgs(); |
|
|
|
dynamic_kernel_list[i]->Execute(); |
|
|
|
@@ -943,6 +942,7 @@ bool KernelRuntime::LaunchKernelMod(const session::KernelGraph &graph) { |
|
|
|
} |
|
|
|
dynamic_kernel_list[i]->PostExecute(); |
|
|
|
} else { |
|
|
|
auto &kernel = kernels[i]; |
|
|
|
auto kernel_mod = AnfAlgo::GetKernelMod(kernel); |
|
|
|
MS_EXCEPTION_IF_NULL(kernel_mod); |
|
|
|
AddressPtrList kernel_inputs; |
|
|
|
|