deviceAllocatorStatus.h

Go to the documentation of this file.

#ifndef HeterogeneousCore_CUDAUtilities_deviceAllocatorStatus_h
#define HeterogeneousCore_CUDAUtilities_deviceAllocatorStatus_h

#include <cstddef>
#include <map>

namespace cms {
  namespace cuda {
    namespace allocator {
      struct TotalBytes {
        // CMS: add explicit std namespace
        std::size_t free;
        std::size_t live;
        std::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