p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

mindir load 2g tensor

This commit is contained in:
lanzhineng 2021-11-11 10:09:08 +08:00
parent 0a54d2adcd
commit a7b0f36c62
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mindspore/core/utils/ms_utils_secure.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2019 Huawei Technologies Co., Ltd
* Copyright 2021 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -17,10 +17,18 @@
#define MINDSPORE_CORE_UTILS_MS_UTILS_SECURE_H_
#include "securec/include/securec.h"
#include "utils/log_adapter.h"
namespace mindspore {
namespace common {
static inline errno_t huge_memcpy_s(uint8_t *dest, size_t destMax, const uint8_t *src, size_t count) {
static inline errno_t huge_memcpy_s(uint8_t *destAddr, size_t destMaxLen, const uint8_t *srcAddr, size_t srcLen) {
MS_EXCEPTION_IF_NULL(destAddr);
MS_EXCEPTION_IF_NULL(srcAddr);
auto dest = destAddr;
auto src = srcAddr;
auto destMax = destMaxLen;
auto count = srcLen;
while (destMax > SECUREC_MEM_MAX_LEN && count > SECUREC_MEM_MAX_LEN) {
auto ret = memcpy_s(dest, SECUREC_MEM_MAX_LEN, src, SECUREC_MEM_MAX_LEN);
if (ret != 0) {