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

[code check]fix fuzz

This commit is contained in:
zhengyuanhua 2022-02-15 14:37:55 +08:00
parent 879b78c140
commit 5f8e89cd2b
3 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mindspore/ccsrc/cxx_api/model/model_converter_utils/multi_process.cc
View File

@ -26,8 +26,15 @@
namespace mindspore {
namespace {
constexpr uint64_t kSharedMemorySize = 100ull << 20; // 100 MB
constexpr int kOneMillisecond = 1000; // 1ms
constexpr int kOneHundredMilliseconds = 100000; // 100ms
constexpr timespec kOneMillisecond = {
0, // 0 seconds
1 * 1000L * 1000L, // And 1 ms
};
constexpr timespec kOneHundredMilliseconds = {
0, // 0 seconds
100 * 1000L * 1000L, // And 100 ms
};
} // namespace
MultiProcess::MultiProcess() = default;
@ -164,7 +171,7 @@ Status MultiProcess::SendMsg(const void *buffer, uint64_t msg_len) {
send_msg_->read_ready_flag = 1;
MS_LOG_INFO << "Send start " << cur_offset << ", msg len " << sub_msg_len << ", total len " << msg_len;
while (!send_msg_->read_finish_flag && !peer_stopped_) {
(void)usleep(kOneMillisecond); // 1ms
(void)nanosleep(&kOneMillisecond, nullptr); // 1ms
}
if (peer_stopped_) {
if (!send_msg_->read_finish_flag) {
@ -185,7 +192,7 @@ Status MultiProcess::ReceiveMsg(const CreateBufferCall &create_buffer_call) {
do {
MS_LOG_INFO << "Receive start from " << cur_offset;
while (!receive_msg_->read_ready_flag && !peer_stopped_) {
(void)usleep(kOneMillisecond); // 1ms
(void)nanosleep(&kOneMillisecond, nullptr); // 1ms
}
if (peer_stopped_) {
return kMEFailed;
@ -232,7 +239,7 @@ void MultiProcess::HeartbeatThreadFuncInner() {
}
}
send_msg_->heartbeat += 1;
(void)usleep(kOneHundredMilliseconds); // sleep 100 ms
(void)nanosleep(&kOneHundredMilliseconds, nullptr); // sleep 100 ms
}
}
} // namespace mindspore

6
mindspore/lite/CMakeLists.txt
View File

@ -401,7 +401,11 @@ elseif(WIN32)
set(RUNTIME_COMPONENT_NAME "win-x64")
endif()
else()
set(RUNTIME_COMPONENT_NAME "linux-x64")
if((CMAKE_SYSTEM_PROCESSOR MATCHES "aarch64") AND MSLITE_ENABLE_ACL)
set(RUNTIME_COMPONENT_NAME "linux-aarch64")
else()
set(RUNTIME_COMPONENT_NAME "linux-x64")
endif()
endif()
string(REPLACE "/mindspore/lite" "" TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR})

5
mindspore/lite/src/runtime/kernel/arm/int8/scale_int8.cc
View File

@ -231,7 +231,10 @@ int ScaleInt8CPUKernel::Prepare() {
return RET_OK;
}
ReSize();
if (ReSize() != RET_OK) {
MS_LOG(ERROR) << "Scale prepare resize failed.";
return RET_ERROR;
}
return RET_OK;
}