OpenMP device manager. More...

Collaboration diagram for OMPDeviceManager:

Public Member Functions
	OMPDeviceManager (int &default_device_num, int &num_devices, int local_rank, int local_size)

Private Attributes
int	omp_default_device_num

const int	omp_device_count

Detailed Description

OpenMP device manager.

Definition at line 22 of file OMPDeviceManager.h.

Constructor & Destructor Documentation

◆ OMPDeviceManager()

OMPDeviceManager	(	int &	default_device_num,
		int &	num_devices,
		int	local_rank,
		int	local_size
	)

Definition at line 23 of file OMPDeviceManager.cpp.

References qmcplusplus::app_warning(), qmcplusplus::determineDefaultDeviceNum(), OMPDeviceManager::omp_default_device_num, and OMPDeviceManager::omp_device_count.

     : omp_default_device_num(-1), omp_device_count(omp_get_num_devices())
 {
   if (num_devices == 0)
     num_devices = omp_device_count;
   else if (num_devices != omp_device_count)
     throw std::runtime_error("Inconsistent number of OpenMP devices with the previous record!");
   if (omp_device_count > local_size)
     app_warning() << "More OpenMP devices than the number of MPI ranks. "
                   << "Some devices will be left idle.\n"
                   << "There is potential performance issue with the GPU affinity.\n";
   if (num_devices > 0)
   {
     omp_default_device_num = determineDefaultDeviceNum(omp_device_count, local_rank, local_size);
     if (default_device_num < 0)
       default_device_num = omp_default_device_num;
     else if (default_device_num != omp_default_device_num)
       throw std::runtime_error("Inconsistent assigned OpenMP devices with the previous record!");
     omp_set_default_device(omp_default_device_num);
   }
 }

Member Data Documentation

◆ omp_default_device_num

int omp_default_device_num

private

Definition at line 24 of file OMPDeviceManager.h.

Referenced by OMPDeviceManager::OMPDeviceManager().

◆ omp_device_count

const int omp_device_count

private

Definition at line 25 of file OMPDeviceManager.h.