diff --git a/source/nvidia/cuda/nvidia-cuda.cpp b/source/nvidia/cuda/nvidia-cuda.cpp index 1b3be61..0eeca6e 100644 --- a/source/nvidia/cuda/nvidia-cuda.cpp +++ b/source/nvidia/cuda/nvidia-cuda.cpp @@ -64,12 +64,14 @@ nvidia::cuda::cuda::cuda() CUDA_LOAD_SYMBOL_V2(cuDevicePrimaryCtxSetFlags); // Context Management + CUDA_LOAD_SYMBOL_V2(cuCtxCreate); CUDA_LOAD_SYMBOL_V2(cuCtxDestroy); CUDA_LOAD_SYMBOL(cuCtxGetCurrent); CUDA_LOAD_SYMBOL(cuCtxGetStreamPriorityRange); CUDA_LOAD_SYMBOL_V2(cuCtxPopCurrent); CUDA_LOAD_SYMBOL_V2(cuCtxPushCurrent); CUDA_LOAD_SYMBOL(cuCtxSetCurrent); + CUDA_LOAD_SYMBOL(cuCtxSynchronize); // Memory Management CUDA_LOAD_SYMBOL_V2(cuArrayGetDescriptor); diff --git a/source/nvidia/cuda/nvidia-cuda.hpp b/source/nvidia/cuda/nvidia-cuda.hpp index eb17dbc..f9b3535 100644 --- a/source/nvidia/cuda/nvidia-cuda.hpp +++ b/source/nvidia/cuda/nvidia-cuda.hpp @@ -160,7 +160,7 @@ namespace nvidia::cuda { CUDA_DEFINE_FUNCTION(cuDevicePrimaryCtxSetFlags, cu_device_t device, cu_context_flags flags); // Context Management - // cuCtxCreate_v2 + CUDA_DEFINE_FUNCTION(cuCtxCreate, cu_context_t* ctx, cu_context_flags flags, cu_device_t device); CUDA_DEFINE_FUNCTION(cuCtxDestroy, cu_context_t ctx); // cuCtxGetApiVersion // cuCtxGetCacheConfig @@ -177,6 +177,7 @@ namespace nvidia::cuda { // cuCtxSetLimit // cuCtxSetSharedMemConfig // cuCtxSynchronize + CUDA_DEFINE_FUNCTION(cuCtxSynchronize); // UNDOCUMENTED? cuCtxResetPersistingL2Cache // Module Management @@ -395,7 +396,7 @@ namespace nvidia::cuda { // cuGraphicsD3D10RegisterResource // Direct3D11 Interopability - CUDA_DEFINE_FUNCTION(cuD3D11GetDevice, std::int32_t* device, IDXGIAdapter* adapter); + CUDA_DEFINE_FUNCTION(cuD3D11GetDevice, cu_device_t* device, IDXGIAdapter* adapter); // cuD3D11GetDevices CUDA_DEFINE_FUNCTION(cuGraphicsD3D11RegisterResource, cu_graphics_resource_t* resource, ID3D11Resource* d3dresource, std::uint32_t flags);