HeterogeneousCore/CUDAUtilities/src/cachingAllocators.cc - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/src/cachingAllocators.cc

16 строк555 B

Matti Kortelainen

Add public interface for constructing and freeing caching allocators

19 авг 2021, 20:28

19 авг 2021, 20:28e5ed64c

#include "HeterogeneousCore/CUDAUtilities/interface/cachingAllocators.h"

#include "getCachingDeviceAllocator.h"
#include "getCachingHostAllocator.h"

namespace cms::cuda::allocator {
  void cachingAllocatorsConstruct() {
    cms::cuda::allocator::getCachingDeviceAllocator();
    cms::cuda::allocator::getCachingHostAllocator();
  }

  void cachingAllocatorsFreeCached() {
    cms::cuda::allocator::getCachingDeviceAllocator().FreeAllCached();
    cms::cuda::allocator::getCachingHostAllocator().FreeAllCached();
  }
}  // namespace cms::cuda::allocator