diff --git a/zluda/src/impl/module.rs b/zluda/src/impl/module.rs index 981ab06..5e8f371 100644 --- a/zluda/src/impl/module.rs +++ b/zluda/src/impl/module.rs @@ -105,6 +105,15 @@ pub(crate) fn get_function( unsafe { hipModuleGetFunction(hfunc, hmod.base, name) } } +pub(crate) fn get_global_v2( + dptr: *mut hipDeviceptr_t, + bytes: *mut usize, + hmod: &Module, + name: *const ::core::ffi::c_char, +) -> hipError_t { + unsafe { hipModuleGetGlobal(dptr, bytes, hmod.base, name) } +} + pub(crate) fn get_loading_mode(mode: &mut cuda_types::cuda::CUmoduleLoadingMode) -> CUresult { *mode = cuda_types::cuda::CUmoduleLoadingMode::CU_MODULE_EAGER_LOADING; Ok(()) diff --git a/zluda/src/lib.rs b/zluda/src/lib.rs index 1fdc1ea..b90d4d6 100644 --- a/zluda/src/lib.rs +++ b/zluda/src/lib.rs @@ -112,6 +112,7 @@ cuda_macros::cuda_function_declarations!( cuMemsetD32_v2, cuMemsetD8_v2, cuModuleGetFunction, + cuModuleGetGlobal_v2, cuModuleGetLoadingMode, cuModuleLoadData, cuModuleUnload,