From cdadcb4a700dd07421222df4aaabc074476beae3 Mon Sep 17 00:00:00 2001 From: kingfo Date: Wed, 15 Jul 2020 18:13:03 +0800 Subject: [PATCH] replace unsafe function --- .../ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc b/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc index 83f7fae582..5c7af1eb59 100644 --- a/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc +++ b/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc @@ -153,7 +153,7 @@ AnfNodePtr TensorMultiplyBase::NewTensorFilledWithData(const AnfNodePtr &node, c char *data = reinterpret_cast(new_tensor_ptr->data_c()); if (x == nullptr) { - std::memset(data, 0, mem_size); + memset_s(data, mem_size, 0, mem_size); auto new_vnode = NewValueNode(new_tensor_ptr); new_vnode->set_abstract(new_tensor_ptr->ToAbstract()); return new_vnode; @@ -188,10 +188,11 @@ AnfNodePtr TensorMultiplyBase::NewTensorFilledWithData(const AnfNodePtr &node, c char *source_data = reinterpret_cast(GetPointerToTensorData(x)); if (x_tensor_ptr->DataSize() == 1) { for (int i = 0; i < new_tensor_ptr->ElementsNum(); i++) { - memcpy(data + i * GetTypeByte(tensor_type_ptr), source_data, GetTypeByte(tensor_type_ptr)); + memcpy_s(data + i * GetTypeByte(tensor_type_ptr), GetTypeByte(tensor_type_ptr), source_data, + GetTypeByte(tensor_type_ptr)); } } else { - memcpy(data, source_data, mem_size); + memcpy_s(data, mem_size, source_data, mem_size); } auto new_vnode = NewValueNode(new_tensor_ptr); new_vnode->set_abstract(new_tensor_ptr->ToAbstract());