HeterogeneousCore/CUDAUtilities/interface/deviceAllocatorStatus.h - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/interface/deviceAllocatorStatus.h

25 строк770 B

Malik Shahzad Muzaffar

Update HeterogeneousCore/CUDAUtilities/interface/deviceAllocatorStatus.h

12 авг 2021, 11:38

12 авг 2021, 11:381d6b8dd

#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; }
      };
      /// Map type of device ordinals to the number of cached bytes cached by each device
      using GpuCachedBytes = std::map<int, TotalBytes>;
    }  // namespace allocator

    allocator::GpuCachedBytes deviceAllocatorStatus();
  }  // namespace cuda
}  // namespace cms

#endif