mirrors/ZLUDA
0
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-08-15 23:09:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Implement cuModuleGetLoadingMode (#381)

Browse source
This commit is contained in:
Violet 2025-06-11 15:54:48 -07:00 • committed by GitHub
commit 1715830d82
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
zluda/src/impl/mod.rs
View file

@ -1,6 +1,10 @@
use cuda_types::cuda::*;
use hip_runtime_sys::*;
use std::{ffi::CStr, mem::{self, ManuallyDrop, MaybeUninit}, ptr};
use std::{
ffi::CStr,
mem::{self, ManuallyDrop, MaybeUninit},
ptr,
};
pub(super) mod context;
pub(super) mod device;
@ -131,6 +135,7 @@ from_cuda_nop!(
cuda_types::cuda::CUdevprop,
CUdevice_attribute,
CUdriverProcAddressQueryResult,
CUmoduleLoadingMode,
CUuuid
);
from_cuda_transmute!(

5
zluda/src/impl/module.rs
View file

@ -53,3 +53,8 @@ pub(crate) fn get_function(
) -> hipError_t {
unsafe { hipModuleGetFunction(hfunc, 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(())
}

1
zluda/src/lib.rs
View file

@ -75,6 +75,7 @@ cuda_base::cuda_function_declarations!(
cuMemsetD32_v2,
cuMemsetD8_v2,
cuModuleGetFunction,
cuModuleGetLoadingMode,
cuModuleLoadData,
cuModuleUnload,
cuPointerGetAttribute,