j2k-cuda-runtime 0.7.1

CUDA codec engine and Driver API runtime for j2k device adapters
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

// SPDX-License-Identifier: MIT OR Apache-2.0

use crate::error::CudaError;

use super::ContextInner;

mod stream_ordered;

impl ContextInner {
    pub(crate) fn resolve_pointer_for_context(
        &self,
        ptr: crate::driver::CuDevicePtr,
    ) -> Result<crate::driver::CuDevicePtr, CudaError> {
        const CU_POINTER_ATTRIBUTE_CONTEXT: i32 = 1;
        let mut pointer_context: crate::driver::CuContext = std::ptr::null_mut();
        self.with_current_resource_operation(|| {
            // SAFETY: CUDA writes one CUcontext value to pointer_context for
            // the live device pointer supplied by the caller.
            self.driver.check("cuPointerGetAttribute", unsafe {
                (self.driver.cu_pointer_get_attribute)(
                    (&raw mut pointer_context).cast(),
                    CU_POINTER_ATTRIBUTE_CONTEXT,
                    ptr,
                )
            })
        })?;
        if !pointer_context.is_null() {
            if pointer_context != self.context {
                return Err(CudaError::InvalidArgument {
                    message: "external CUDA pointer belongs to a different context".to_string(),
                });
            }
            return Ok(ptr);
        }
        stream_ordered::resolve(self, ptr)
    }
}