Implement cuModuleGetGlobal_v2 (#454)

This commit is contained in:
Violet
2025-07-30 16:34:21 -07:00
committed by GitHub
parent 66db19a061
commit 4d163a4d9b
2 changed files with 10 additions and 0 deletions

View File

@ -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(())

View File

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