From 9d6cfba5c6d37391ffaca0a6a2b26c04c4f52912 Mon Sep 17 00:00:00 2001 From: xulei Date: Mon, 17 Oct 2022 15:36:39 +0800 Subject: [PATCH] fix non-check return of safe function --- .../aicpu/aicpu_ops/random_choice_with_mask_kernels.cc | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/random_choice_with_mask_kernels.cc b/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/random_choice_with_mask_kernels.cc index f02281ef552..c80e6ceb7eb 100644 --- a/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/random_choice_with_mask_kernels.cc +++ b/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/random_choice_with_mask_kernels.cc @@ -232,7 +232,14 @@ uint32_t RandomChoiceWithMaskKernel::DoCompute() { return kAicpuKernelStateInternalError; } size_t copy_output_bytes = IntToSize(new_output_length) * sizeof(int32_t); - (void)memcpy_s(output_coordinate, copy_output_bytes, output, copy_output_bytes); + if (memcpy_s(output_coordinate, copy_output_bytes, output, copy_output_bytes) != EOK) { + AICPU_LOGE("memcpy_s memory failed!"); + free(input_dim); + free(mask_dim); + free(tmp_output); + free(output); + return kAicpuKernelStateInternalError; + } UpdateOutput(dims_, non_zero_num, count_, output_length, mask_dim, output_coordinate, mask); AICPU_LOGI("no zero num is %d, output_length is %d ", non_zero_num, output_length); UpdateOutputShapeValue(non_zero_num, output_length);