Implement cuModuleGetGlobal_v2

This commit is contained in:
Violet
2025-07-25 07:25:30 +00:00
parent b8bcbec295
commit 0bb4f10b51
2 changed files with 10 additions and 0 deletions

View File

@ -103,6 +103,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(())

View File

@ -112,6 +112,7 @@ cuda_macros::cuda_function_declarations!(
cuMemsetD32_v2,
cuMemsetD8_v2,
cuModuleGetFunction,
cuModuleGetGlobal_v2,
cuModuleGetLoadingMode,
cuModuleLoadData,
cuModuleUnload,