| 28 | } |
| 29 | |
| 30 | CUDAExternalBuffer::CUDAExternalBuffer(Engine* engine, |
| 31 | ExternalMemoryTypeFlags handleType, |
| 32 | void* handle, const void* name, size_t byteSize) |
| 33 | : USMBuffer(engine) |
| 34 | { |
| 35 | cudaExternalMemoryHandleDesc handleDesc{}; |
| 36 | handleDesc.handle.win32.handle = handle; |
| 37 | handleDesc.handle.win32.name = name; |
| 38 | handleDesc.size = byteSize; |
| 39 | |
| 40 | if (handleType & ExternalMemoryTypeFlag::Dedicated) |
| 41 | { |
| 42 | handleDesc.flags = cudaExternalMemoryDedicated; |
| 43 | handleType ^= ExternalMemoryTypeFlag::Dedicated; |
| 44 | } |
| 45 | |
| 46 | if (handleType == ExternalMemoryTypeFlag::OpaqueWin32) |
| 47 | handleDesc.type = cudaExternalMemoryHandleTypeOpaqueWin32; |
| 48 | else if (handleType == ExternalMemoryTypeFlag::OpaqueWin32KMT) |
| 49 | handleDesc.type = cudaExternalMemoryHandleTypeOpaqueWin32Kmt; |
| 50 | else if (handleType == ExternalMemoryTypeFlag::D3D11Texture || |
| 51 | handleType == ExternalMemoryTypeFlag::D3D11Resource) |
| 52 | { |
| 53 | handleDesc.type = cudaExternalMemoryHandleTypeD3D11Resource; |
| 54 | handleDesc.flags = cudaExternalMemoryDedicated; // mandatory |
| 55 | } |
| 56 | else if (handleType == ExternalMemoryTypeFlag::D3D11TextureKMT || |
| 57 | handleType == ExternalMemoryTypeFlag::D3D11ResourceKMT) |
| 58 | { |
| 59 | handleDesc.type = cudaExternalMemoryHandleTypeD3D11ResourceKmt; |
| 60 | handleDesc.flags = cudaExternalMemoryDedicated; // mandatory |
| 61 | } |
| 62 | else if (handleType == ExternalMemoryTypeFlag::D3D12Heap) |
| 63 | handleDesc.type = cudaExternalMemoryHandleTypeD3D12Heap; |
| 64 | else if (handleType == ExternalMemoryTypeFlag::D3D12Resource) |
| 65 | { |
| 66 | handleDesc.type = cudaExternalMemoryHandleTypeD3D12Resource; |
| 67 | handleDesc.flags = cudaExternalMemoryDedicated; // mandatory |
| 68 | } |
| 69 | else |
| 70 | throw Exception(Error::InvalidArgument, "external memory type not supported by the device"); |
| 71 | |
| 72 | init(handleDesc); |
| 73 | } |
| 74 | |
| 75 | void CUDAExternalBuffer::init(const cudaExternalMemoryHandleDesc& handleDesc) |
| 76 | { |