| 73 | } |
| 74 | |
| 75 | void CUDAExternalBuffer::init(const cudaExternalMemoryHandleDesc& handleDesc) |
| 76 | { |
| 77 | checkError(cudaImportExternalMemory(&extMem, &handleDesc)); |
| 78 | |
| 79 | void* devPtr = nullptr; |
| 80 | cudaExternalMemoryBufferDesc bufferDesc{}; |
| 81 | bufferDesc.offset = 0; |
| 82 | bufferDesc.size = handleDesc.size; |
| 83 | bufferDesc.flags = 0; |
| 84 | checkError(cudaExternalMemoryGetMappedBuffer(&devPtr, extMem, &bufferDesc)); |
| 85 | |
| 86 | ptr = static_cast<char*>(devPtr); |
| 87 | byteSize = handleDesc.size; |
| 88 | shared = true; |
| 89 | storage = Storage::Device; |
| 90 | } |
| 91 | |
| 92 | CUDAExternalBuffer::~CUDAExternalBuffer() |
| 93 | { |
nothing calls this directly
no test coverage detected