cms::cuda::ScopedSetDevice Class Reference

#include <ScopedSetDevice.h>

Public Member Functions
	ScopedSetDevice ()
	ScopedSetDevice (int device)
void	set (int device)
	~ScopedSetDevice ()

Private Attributes
int	originalDevice_

Detailed Description

Definition at line 10 of file ScopedSetDevice.h.

Constructor & Destructor Documentation

ScopedSetDevice() [1/2]

cms::cuda::ScopedSetDevice::ScopedSetDevice ( )

inline

Definition at line 13 of file ScopedSetDevice.h.

References cudaCheck, and originalDevice_.

                        {
        // Store the original device
        cudaCheck(cudaGetDevice(&originalDevice_));
      }

ScopedSetDevice() [2/2]

cms::cuda::ScopedSetDevice::ScopedSetDevice ( int  device )

inlineexplicit

Definition at line 19 of file ScopedSetDevice.h.

                                           : ScopedSetDevice() {
        // Change the current device
        set(device);
      }

~ScopedSetDevice()

cms::cuda::ScopedSetDevice::~ScopedSetDevice ( )

inline

Definition at line 25 of file ScopedSetDevice.h.

References originalDevice_.

                         {
        // Intentionally don't check the return value to avoid
        // exceptions to be thrown. If this call fails, the process is
        // doomed anyway.
        cudaSetDevice(originalDevice_);
      }

Member Function Documentation

set()

void cms::cuda::ScopedSetDevice::set ( int  device )

inline

Definition at line 34 of file ScopedSetDevice.h.

References cudaCheck.

Referenced by cms::cuda::ESProduct< HcalGainWidthsGPU::Product >::ESProduct().

                           {
        // Change the current device
        cudaCheck(cudaSetDevice(device));
      }

Member Data Documentation

originalDevice_

int cms::cuda::ScopedSetDevice::originalDevice_

private

Definition at line 40 of file ScopedSetDevice.h.

Referenced by ScopedSetDevice(), and ~ScopedSetDevice().