diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/apply_power_sign_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/apply_power_sign_cpu_kernel.cc index 05b0f4c07b8..39919bb7fbc 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/apply_power_sign_cpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/apply_power_sign_cpu_kernel.cc @@ -83,6 +83,7 @@ void ApplyPowerSignCpuKernelMod::LaunchPowerSign(const std::vector &inputs, const std::vector &outputs) { + MS_EXCEPTION_IF_NULL(base_operator); kernel_name_ = base_operator->name(); dtype_ = inputs[0]->GetDtype(); batch_rank_ = base_operator->get_batch_rank(); diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/scale_grad_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/scale_grad_cpu_kernel.cc index 10210fd80ab..b53f793ecf1 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/scale_grad_cpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/scale_grad_cpu_kernel.cc @@ -54,6 +54,7 @@ void ScaleGradCpuKernelMod::LaunchScaleGradPerGrad(const std::vector if (scale_addr_half != nullptr) { x1 = static_cast(*scale_addr_half); } else { + MS_EXCEPTION_IF_NULL(scale_addr_float); x1 = static_cast(*scale_addr_float); } @@ -101,7 +102,9 @@ std::vector ScaleGradCpuKernelMod::GetOpSupport() { bool ScaleGradCpuKernelMod::Init(const BaseOperatorPtr &base_operator, const std::vector &inputs, const std::vector &outputs) { + MS_EXCEPTION_IF_NULL(base_operator); auto kernel_ptr = std::dynamic_pointer_cast(base_operator); + MS_EXCEPTION_IF_NULL(kernel_ptr); kernel_name_ = kernel_ptr->name(); auto input_size = inputs.size(); for (size_t index = 0; index < input_size; index++) { diff --git a/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/scalegrad_gpu_kernel.cc b/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/scalegrad_gpu_kernel.cc index a6dfab83322..52ca3bccd66 100644 --- a/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/scalegrad_gpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/scalegrad_gpu_kernel.cc @@ -50,6 +50,7 @@ void ScaleGradGpuKernelMod::LaunchScaleGradPerGrad(const std::vector ScaleGradKernel(outputs[index]->size / sizeof(T), input_addr, *scale_addr_half, output_addr, reinterpret_cast(stream_ptr)); } else { + MS_EXCEPTION_IF_NULL(scale_addr_float); ScaleGradKernel(outputs[index]->size / sizeof(T), input_addr, *scale_addr_float, output_addr, reinterpret_cast(stream_ptr)); } @@ -84,7 +85,9 @@ bool ScaleGradGpuKernelMod::Launch(const std::vector &inputs, const bool ScaleGradGpuKernelMod::Init(const BaseOperatorPtr &base_operator, const std::vector &inputs, const std::vector &outputs) { + MS_EXCEPTION_IF_NULL(base_operator); auto kernel_ptr = std::dynamic_pointer_cast(base_operator); + MS_EXCEPTION_IF_NULL(kernel_ptr); kernel_name_ = kernel_ptr->name(); auto input_size = inputs.size(); for (size_t index = 0; index < input_size; index++) {