deviceAllocatorStatus.h

Go to the documentation of this file.

#ifndef HeterogeneousCore_CUDAUtilities_deviceAllocatorStatus_h
#define HeterogeneousCore_CUDAUtilities_deviceAllocatorStatus_h
 
#include <map>
 
namespace cms {
  namespace cuda {
    namespace allocator {
      struct TotalBytes {
        size_t free;
        size_t live;
        size_t liveRequested;  // CMS: monitor also requested amount
        TotalBytes() { free = live = liveRequested = 0; }
      };
      using GpuCachedBytes = std::map<int, TotalBytes>;
    }  // namespace allocator
 
    allocator::GpuCachedBytes deviceAllocatorStatus();
  }  // namespace cuda
}  // namespace cms
 
#endif