mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!44015 fix non-check return of safe function 

Merge pull request !44015 from xulei/clean_code
This commit is contained in:
i-robot 2022-10-18 08:47:27 +00:00 committed by Gitee
parent ceb91c5f8f 9d6cfba5c6
commit 0c0249b243
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/random_choice_with_mask_kernels.cc
View File

@ -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);