n_threads bug

4 years ago · 8b63e5a047
parent 95dd988807
commit 8b63e5a047
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/quantum/pqc_cpu_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/quantum/pqc_cpu_kernel.cc
@ -175,7 +175,7 @@ void PQCCPUKernel::InitKernel(const CNodePtr &kernel_node) {

  hams_ = mindquantum::transformer::HamiltoniansTransfor(hams_pauli_coeff_, hams_pauli_word_, hams_pauli_qubit_);

-  n_threads_user_ = common::ThreadPool::GetInstance().GetSyncRunThreadNum();
+  n_threads_user_ = std::min(n_threads_user_, common::ThreadPool::GetInstance().GetSyncRunThreadNum());
  if (n_samples_ < n_threads_user_) {
    n_threads_user_ = n_samples_;
  }