allocate_device.h

Go to the documentation of this file.

#ifndef HeterogeneousCore_CUDAUtilities_allocate_device_h
#define HeterogeneousCore_CUDAUtilities_allocate_device_h
 
#include <cuda_runtime.h>
 
namespace cms {
  namespace cuda {
    // Allocate device memory
    void *allocate_device(int dev, size_t nbytes, cudaStream_t stream);
 
    // Free device memory (to be called from unique_ptr)
    void free_device(int device, void *ptr);
  }  // namespace cuda
}  // namespace cms
 
#endif