device_wrapper_gpu.cu 509 B

1234567891011121314151617181920
  1. #include <stdio.h>
  2. #include <assert.h>
  3. extern "C" {
  4. void* host_malloc_cuda(size_t size){
  5. void* p;
  6. cudaError_t error = cudaHostAlloc(&p, size, cudaHostAllocPortable);
  7. if (error != cudaSuccess) fprintf(stderr,"CUDA Error: %s \n", cudaGetErrorString(error));
  8. assert(error == cudaSuccess);
  9. return p;
  10. }
  11. void host_free_cuda(void* p){
  12. cudaError_t error = cudaFreeHost(p);
  13. if (error != cudaSuccess) fprintf(stderr,"CUDA Error: %s \n", cudaGetErrorString(error));
  14. assert(error == cudaSuccess);
  15. }
  16. }