Applied-Software/ZLUDA
5
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-07-18 17:56:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add device getter

Browse Source
This commit is contained in:
Andrzej Janik
2020-02-16 17:13:42 +01:00
parent 6d748a3959
commit 21d091a47d
2 changed files with 38 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
notcuda/src/cu.rs
View File

@ -85,6 +85,7 @@ impl Result {
l0::ze_result_t::ZE_RESULT_SUCCESS => Result::SUCCESS, l0::ze_result_t::ZE_RESULT_SUCCESS => Result::SUCCESS,
l0::ze_result_t::ZE_RESULT_ERROR_UNINITIALIZED => Result::ERROR_NOT_INITIALIZED, l0::ze_result_t::ZE_RESULT_ERROR_UNINITIALIZED => Result::ERROR_NOT_INITIALIZED,
l0::ze_result_t::ZE_RESULT_ERROR_INVALID_ENUMERATION => Result::ERROR_INVALID_VALUE, l0::ze_result_t::ZE_RESULT_ERROR_INVALID_ENUMERATION => Result::ERROR_INVALID_VALUE,
l0::ze_result_t::ZE_RESULT_ERROR_INVALID_ARGUMENT => Result::ERROR_INVALID_VALUE,
l0::ze_result_t::ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY => Result::ERROR_OUT_OF_MEMORY, l0::ze_result_t::ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY => Result::ERROR_OUT_OF_MEMORY,
_ => Result::ERROR_UNKNOWN _ => Result::ERROR_UNKNOWN
} }
@ -96,7 +97,3 @@ impl Result {
pub struct Uuid { pub struct Uuid {
pub x: [std::os::raw::c_uchar; 16] pub x: [std::os::raw::c_uchar; 16]
} }
pub struct Device {
base: level_zero_sys::ze_driver_handle_t
}

41
notcuda/src/lib.rs
View File

@ -37,6 +37,39 @@ impl Driver {
l0_check!{ l0::zeDriverGet(&mut driver_count, &mut handle) }; l0_check!{ l0::zeDriverGet(&mut driver_count, &mut handle) };
Ok(Driver{ base: handle }) Ok(Driver{ base: handle })
} }
fn call<F: FnOnce(&mut Driver) -> l0::ze_result_t>(f: F) -> cu::Result {
let mut lock = GLOBAL_STATE.try_lock();
if let Ok(ref mut mutex) = lock {
match **mutex {
None => return cu::Result::ERROR_NOT_INITIALIZED,
Some(ref mut driver) => {
return cu::Result::from_l0(f(driver));
}
}
} else {
return cu::Result::ERROR_UNKNOWN;
}
}
fn device_get_count(&self, count: *mut i32) -> l0::ze_result_t {
unsafe { l0::zeDeviceGet(self.base, count as *mut _ as *mut _, ptr::null_mut()) }
}
fn device_get(&self, device: *mut l0::ze_device_handle_t, ordinal: ::std::os::raw::c_int) -> l0::ze_result_t {
let count = (ordinal as u32) + 1;
let mut devices_found = count;
let mut handles = vec![ptr::null_mut(); count as usize];
let result = unsafe { l0::zeDeviceGet(self.base, &mut devices_found, handles.as_mut_ptr()) };
if result != l0::ze_result_t::ZE_RESULT_SUCCESS {
return result;
}
if devices_found < count {
return l0::ze_result_t::ZE_RESULT_ERROR_INVALID_ARGUMENT;
}
unsafe { *device = handles[(count as usize) - 1] };
l0::ze_result_t::ZE_RESULT_SUCCESS
}
} }
#[no_mangle] #[no_mangle]
@ -66,11 +99,11 @@ pub unsafe extern "stdcall" fn cuInit(_: *const std::os::raw::c_uint) -> cu::Res
} }
#[no_mangle] #[no_mangle]
pub extern "stdcall" fn cuDeviceGetCount(count: &mut std::os::raw::c_int) -> cu::Result { pub extern "stdcall" fn cuDeviceGetCount(count: *mut std::os::raw::c_int) -> cu::Result {
return cu::Result::SUCCESS; Driver::call(|driver| driver.device_get_count(count))
} }
#[no_mangle] #[no_mangle]
pub extern "stdcall" fn cuDeviceGet(device: *mut cu::Device, ordinal: ::std::os::raw::c_int) -> cu::Result { pub extern "stdcall" fn cuDeviceGet(device: *mut l0::ze_device_handle_t, ordinal: ::std::os::raw::c_int) -> cu::Result {
unimplemented!() Driver::call(|driver| driver.device_get(device, ordinal))
} }