Browse Source

!3097 replace unsafe function

Merge pull request !3097 from wangqiuliang/replace-unsafe-function
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d07da62490
1 changed files with 4 additions and 3 deletions
  1. +4
    -3
      mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc

+ 4
- 3
mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc View File

@@ -153,7 +153,7 @@ AnfNodePtr TensorMultiplyBase::NewTensorFilledWithData(const AnfNodePtr &node, c
char *data = reinterpret_cast<char *>(new_tensor_ptr->data_c()); char *data = reinterpret_cast<char *>(new_tensor_ptr->data_c());


if (x == nullptr) { if (x == nullptr) {
std::memset(data, 0, mem_size);
memset_s(data, mem_size, 0, mem_size);
auto new_vnode = NewValueNode(new_tensor_ptr); auto new_vnode = NewValueNode(new_tensor_ptr);
new_vnode->set_abstract(new_tensor_ptr->ToAbstract()); new_vnode->set_abstract(new_tensor_ptr->ToAbstract());
return new_vnode; return new_vnode;
@@ -188,10 +188,11 @@ AnfNodePtr TensorMultiplyBase::NewTensorFilledWithData(const AnfNodePtr &node, c
char *source_data = reinterpret_cast<char *>(GetPointerToTensorData(x)); char *source_data = reinterpret_cast<char *>(GetPointerToTensorData(x));
if (x_tensor_ptr->DataSize() == 1) { if (x_tensor_ptr->DataSize() == 1) {
for (int i = 0; i < new_tensor_ptr->ElementsNum(); i++) { 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 { } else {
memcpy(data, source_data, mem_size);
memcpy_s(data, mem_size, source_data, mem_size);
} }
auto new_vnode = NewValueNode(new_tensor_ptr); auto new_vnode = NewValueNode(new_tensor_ptr);
new_vnode->set_abstract(new_tensor_ptr->ToAbstract()); new_vnode->set_abstract(new_tensor_ptr->ToAbstract());


Loading…
Cancel
Save