#include <kaldi-thread.h>

Collaboration diagram for MultiThreader< C >:

Public Member Functions
	MultiThreader (int32 num_threads, const C &c_in)

	~MultiThreader ()

Private Attributes
std::vector< std::thread >	threads_

std::vector< C >	cvec_

Detailed Description

template<class C>
class kaldi::MultiThreader< C >

Definition at line 117 of file kaldi-thread.h.

Constructor & Destructor Documentation

◆ MultiThreader()

MultiThreader	(	int32	num_threads,
		const C &	c_in
	)

inline

Definition at line 119 of file kaldi-thread.h.

References rnnlm::i.

                                                   :
     threads_(std::max<int32>(1, num_threads)),
     cvec_(std::max<int32>(1, num_threads), c_in) {
     if (num_threads == 0) {
       // This is a special case with num_threads == 0, which behaves like with
       // num_threads == 1 but without creating extra threads.  This can be
       // useful in GPU computations where threads cannot be used.
       cvec_[0].thread_id_ = 0;
       cvec_[0].num_threads_ = 1;
       (cvec_[0])();
     } else {
       for (int32 i = 0; i < threads_.size(); i++) {
         cvec_[i].thread_id_ = i;
         cvec_[i].num_threads_ = threads_.size();
         threads_[i] = std::thread(std::ref(cvec_[i]));
       }
     }
   }

◆ ~MultiThreader()

~MultiThreader ( )

inline

Definition at line 137 of file kaldi-thread.h.

References rnnlm::i.

                    {
     for (size_t i = 0; i < threads_.size(); i++)
       if (threads_[i].joinable())
         threads_[i].join();
   }

Member Data Documentation

◆ cvec_

std::vector<C> cvec_

private

Definition at line 144 of file kaldi-thread.h.

◆ threads_

std::vector<std::thread> threads_

private

Definition at line 143 of file kaldi-thread.h.

The documentation for this class was generated from the following file:

util/kaldi-thread.h

Public Member Functions

Private Attributes