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

HeterogeneousCore/CUDAUtilities/interface/cachingAllocators.h

13 строк410 B

Matti Kortelainen

Add public interface for constructing and freeing caching allocators

19 авг 2021, 20:28

19 авг 2021, 20:28e5ed64c

#ifndef HeterogeneousCore_CUDAUtilities_interface_cachingAllocators_h
#define HeterogeneousCore_CUDAUtilities_interface_cachingAllocators_h

namespace cms::cuda::allocator {
  // Use caching or not
  constexpr bool useCaching = true;

  // these intended to be called only from CUDAService
  void cachingAllocatorsConstruct();
  void cachingAllocatorsFreeCached();
}  // namespace cms::cuda::allocator

#endif