!33582 [lite]fix thread-distribution question

Merge pull request !33582 from 徐安越/master1
2022-04-27 03:01:43 +00:00 · 2022-04-27 03:01:43 +00:00 · 42168e9dc9
parent ed4102c960 b096e53779
commit 42168e9dc9
1 changed files with 1 additions and 0 deletions
--- a/mindspore/lite/src/runtime/kernel/cpu/fp32/matmul_fp32_base.cc
+++ b/mindspore/lite/src/runtime/kernel/cpu/fp32/matmul_fp32_base.cc
@ -380,6 +380,7 @@ int MatmulFp32BaseCPUKernel::GetThreadCuttingPolicy() {
      split_points_.push_back(split_point * col_min_unit_);
      split_point += block_col_unit;
    }
+    thread_count_ = split_points_.size();
    parallel_fun_ = &MatmulFp32BaseCPUKernel::ParallelRunByOC;
  }
  return RET_OK;