mirrors/ZLUDA
0
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-07-29 12:28:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Refactor unused code, add residency control

Browse source
This commit is contained in:
Andrzej Janik 2020-07-31 01:48:03 +02:00
commit 9ed3dc54f2
3 changed files with 70 additions and 220 deletions
Download patch file Download diff file Expand all files Collapse all files

19
level_zero/src/ze.rs
View file

@ -708,10 +708,10 @@ impl<'a> Kernel<'a> {
Self(x, PhantomData)
}
pub fn new(module: &'a Module, name: &CStr) -> Result<Self> {
pub fn new_resident(module: &'a Module, name: &CStr) -> Result<Self> {
let desc = sys::ze_kernel_desc_t {
version: sys::ze_kernel_desc_version_t::ZE_KERNEL_DESC_VERSION_CURRENT,
flags: sys::ze_kernel_flag_t::ZE_KERNEL_FLAG_NONE,
flags: sys::ze_kernel_flag_t::ZE_KERNEL_FLAG_FORCE_RESIDENCY,
pKernelName: name.as_ptr() as *const _,
};
let mut result = ptr::null_mut();
@ -719,6 +719,21 @@ impl<'a> Kernel<'a> {
Ok(Self(result, PhantomData))
}
pub fn set_attribute_bool(
&mut self,
attr: sys::ze_kernel_attribute_t,
value: bool,
) -> Result<()> {
let ze_bool: sys::ze_bool_t = if value { 1 } else { 0 };
check!(sys::zeKernelSetAttribute(
self.0,
attr,
mem::size_of::<sys::ze_bool_t>() as u32,
&ze_bool as *const _ as *const _
));
Ok(())
}
pub fn set_arg_buffer<T: 'a, Buff: Into<BufferPtr<'a, T>>>(
&self,
index: u32,