From 283984216ef0d9dae9c6545a596cd1aeabbbaf50 Mon Sep 17 00:00:00 2001 From: ZhengQihao3f3f3f Date: Thu, 29 Apr 2021 16:03:54 +0800 Subject: [PATCH] Modify ops 1 --- .../backend/kernel_compiler/cpu/maximum_grad_cpu_kernel.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/maximum_grad_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/maximum_grad_cpu_kernel.cc index e264996d76..f479b2d3f4 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/maximum_grad_cpu_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/maximum_grad_cpu_kernel.cc @@ -113,8 +113,10 @@ void MaximumGradCPUKernel::LaunchKernel(const std::vector &inputs, c size_t x_tensor_len = GetTensorLen(x_shape_); size_t y_tensor_len = GetTensorLen(y_shape_); - memset(dx_addr, 0, x_tensor_len * sizeof(T)); - memset(dy_addr, 0, y_tensor_len * sizeof(T)); + size_t x_tensor_size = x_tensor_len * sizeof(T); + size_t y_tensor_size = y_tensor_len * sizeof(T); + memset_s(dx_addr, x_tensor_size, 0, x_tensor_size); + memset_s(dy_addr, y_tensor_size, 0, y_tensor_size); std::vector x_shape(dout_shape.size(), 1); std::vector y_shape(dout_shape.size(), 1);