!6750 fix infer tensor compile error

Merge pull request !6750 from kisnwang/r1.0
5 years ago · a6263d2837
parent fafc7ee919 a8e7e283a8
commit a6263d2837
1 changed files with 2 additions and 1 deletions
--- a/include/infer_tensor.h
+++ b/include/infer_tensor.h
@ -76,7 +76,8 @@ class InferTensorBase {
    }
    auto ret = memcpy_s(mutable_data(), data_size(), data, data_len);
    if (ret != 0) {
-      MS_LOG(EXCEPTION) << "Set data memcpy_s failed";
+      MSI_LOG_ERROR << "Set data memcpy_s failed";
+      return false;
    }
    return true;
  }