Project CMSSW displayed by LXR CMSSW_14_1_X_2024-07-25-2300/​HeterogeneousCore/​CUDAUtilities/​interface/​ScopedSetDevice.h	Source navigation Diff markup Identifier search General search
	Version: CMSSW_14_1_X_2024-07-25-2300 CMSSW_14_1_X_2024-07-24-2300 CMSSW_14_1_X_2024-07-23-2300 CMSSW_14_1_X_2024-07-22-2300 CMSSW_14_1_X_2024-07-21-2300 CMSSW_14_1_X_2024-07-19-2300 Revert   Change

File indexing completed on 2024-04-06 12:15:45

 #ifndef HeterogeneousCore_CUDAUtilities_ScopedSetDevice_h
 #define HeterogeneousCore_CUDAUtilities_ScopedSetDevice_h
 
 #include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h"
 
 #include <cuda_runtime.h>
 
 namespace cms {
   namespace cuda {
     class ScopedSetDevice {
     public:
       // Store the original device, without setting a new one
       ScopedSetDevice() {
         // Store the original device
         cudaCheck(cudaGetDevice(&originalDevice_));
       }
 
       // Store the original device, and set a new current device
       explicit ScopedSetDevice(int device) : ScopedSetDevice() {
         // Change the current device
         set(device);
       }
 
       // Restore the original device
       ~ScopedSetDevice() {
         // Intentionally don't check the return value to avoid
         // exceptions to be thrown. If this call fails, the process is
         // doomed anyway.
         cudaSetDevice(originalDevice_);
       }
 
       // Set a new current device, without changing the original device
       // that will be restored when this object is destroyed
       void set(int device) {
         // Change the current device
         cudaCheck(cudaSetDevice(device));
       }
 
     private:
       int originalDevice_;
     };
   }  // namespace cuda
 }  // namespace cms
 
 #endif

This page was automatically generated by the 2.2.1 LXR engine. The LXR team