Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 12:49:26

0001 #ifndef HeterogeneousCore_CUDAUtilities_allocate_device_h
0002 #define HeterogeneousCore_CUDAUtilities_allocate_device_h
0003 
0004 #include <cuda_runtime.h>
0005 
0006 namespace cms {
0007   namespace cuda {
0008     // Allocate device memory
0009     void *allocate_device(int dev, size_t nbytes, cudaStream_t stream);
0010 
0011     // Free device memory (to be called from unique_ptr)
0012     void free_device(int device, void *ptr);
0013   }  // namespace cuda
0014 }  // namespace cms
0015 
0016 #endif