singe-npp 0.1.0-alpha.8

Safe Rust wrappers for NVIDIA Performance Primitives library (NPP).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use singe_cuda::memory::DeviceMemory;

use crate::{
    context::StreamContext,
    error::{Error, Result},
};

pub(crate) fn validate_device_memory_stream_context<T>(
    stream_context: &StreamContext,
    memory: &DeviceMemory<T>,
) -> Result<()> {
    if memory.is_empty() {
        return Ok(());
    }

    let attributes = DeviceMemory::pointer_attributes(memory.as_ptr())?;
    if attributes.device != stream_context.device_id() {
        return Err(Error::StreamContextMismatch);
    }

    Ok(())
}