!13179 n_threads bug in PQC operator

From: @donghufeng
Reviewed-by: @zhoufeng54,@kisnwang
Signed-off-by: @kisnwang
pull/13179/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit e07a4f820d

@ -175,7 +175,7 @@ void PQCCPUKernel::InitKernel(const CNodePtr &kernel_node) {
hams_ = mindquantum::transformer::HamiltoniansTransfor(hams_pauli_coeff_, hams_pauli_word_, hams_pauli_qubit_); 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_) { if (n_samples_ < n_threads_user_) {
n_threads_user_ = n_samples_; n_threads_user_ = n_samples_;
} }

Loading…
Cancel
Save