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

Fix minor problems with a private CUDA function

Browse Source
This commit is contained in:
Andrzej Janik
2020-09-24 02:20:54 +02:00
parent 3f41f21acb
commit 42bcd999eb
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
notcuda/src/impl/export_table.rs
View File

@ -323,6 +323,9 @@ fn context_local_storage_ctor_impl(
if cu_ctx == ptr::null_mut() {
context::get_current(&mut cu_ctx)?;
}
if cu_ctx == ptr::null_mut() {
return Err(CUresult::CUDA_ERROR_INVALID_VALUE);
}
unsafe { &*cu_ctx }
.as_ref()
.ok_or(CUresult::CUDA_ERROR_INVALID_CONTEXT)
@ -354,9 +357,12 @@ unsafe extern "C" fn context_local_storage_get_state(
fn context_local_storage_get_state_impl(
ctx_state: *mut *mut cuda_impl::rt::ContextState,
cu_ctx: *mut context::Context,
mut cu_ctx: *mut context::Context,
_: *mut cuda_impl::rt::ContextStateManager,
) -> Result<(), CUresult> {
if cu_ctx == ptr::null_mut() {
context::get_current(&mut cu_ctx)?;
}
if cu_ctx == ptr::null_mut() {
return Err(CUresult::CUDA_ERROR_INVALID_VALUE);
}
@ -369,6 +375,10 @@ fn context_local_storage_get_state_impl(
.map_err(|_| CUresult::CUDA_ERROR_ILLEGAL_STATE)
.map(|mutable| mutable.cuda_state)
})?;
if cuda_state == ptr::null_mut() {
Err(CUresult::CUDA_ERROR_INVALID_VALUE)
} else {
unsafe { *ctx_state = cuda_state };
Ok(())
}
}