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

!8343 [MSLITE][Develop] fix code 

From: @sunsuodong
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2020-11-09 14:29:07 +08:00 committed by Gitee
parent 49e3aa35a2 3ba218ef25
commit 5708bae7e7
4 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/lite/nnacl/minimal_filtering_generator.c
View File

@ -257,6 +257,9 @@ void MatrixMultiplyVec(const float32x4_t *matrix_a, const float32x4_t *matrix_b,
int WinogradWeightTransform(const float *weight_data, float *winograd_data, float *matrix_g, const float *matrix_gt,
int oc_block, int input_unit, int kernel_unit, int channel, int batch, bool pack) {
if (oc_block == 0) {
return NNACL_PARAM_INVALID;
}
// original weight format : ohwi
int oc_block_num = UP_DIV(batch, oc_block);
int block_stride = channel * oc_block;

2
mindspore/lite/src/runtime/kernel/arm/fp32_grad/sparse_softmax_cross_entropy_with_logits.cc
View File

@ -101,7 +101,7 @@ int SparseSoftmaxCrossEntropyWithLogitsCPUKernel::Execute(int task_id) {
Softmax(ins, losses_, sum_data_, &sm_params_);
if (is_train()) {
GradPostExecute(labels, losses_, grads, out);
} else if (out != nullptr) {
} else {
ForwardPostExecute(labels, losses_, out);
}
return RET_OK;

4
mindspore/lite/src/runtime/thread_pool.c
View File

@ -835,6 +835,10 @@ ThreadPool *CreateThreadPool(int thread_num, int mode) {
}
#endif
ThreadPool *thread_pool = (struct ThreadPool *)(malloc(sizeof(ThreadPool)));
if (thread_pool == NULL) {
LOG_ERROR("Malloc ThreadPool failed");
return NULL;
}
thread_pool->thread_num = thread_num > MAX_THREAD_NUM ? MAX_THREAD_NUM : thread_num;
thread_pool->is_alive = ATOMIC_VAR_INIT(true);
thread_pool->mode = mode;

2
mindspore/lite/tools/benchmark/benchmark.cc
View File

@ -275,7 +275,7 @@ int Benchmark::CompareStringData(const std::string &name, tensor::MSTensor *tens
std::cout << " " << output_strings[i] << std::endl;
}
if (calib_strings[i] != output_strings[i]) {
MS_LOG(ERROR) << "";
MS_LOG(ERROR) << "Compare failed, index: " << i;
return RET_ERROR;
}
}