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
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use singe_cuda::memory::DeviceMemory;

use crate::{
    error::{Error, Result},
    types::Size,
};

pub(super) fn validate_same_size(expected: Size, actual: Size) -> Result<()> {
    if expected == actual {
        return Ok(());
    }
    Err(Error::SizeMismatch {
        name: "image size".into(),
        expected,
        actual,
    })
}

pub(super) fn validate_device_constant_len(actual: usize, expected: usize) -> Result<()> {
    if actual >= expected {
        return Ok(());
    }
    Err(Error::LengthMismatch {
        name: "device constants".into(),
        expected,
        actual,
    })
}

pub(super) fn device_constant_ptr<T>(
    constants: &DeviceMemory<T>,
    expected: usize,
) -> Result<*const T> {
    validate_device_constant_len(constants.len(), expected)?;
    Ok(constants.as_ptr())
}

pub(super) fn device_constant_mut_ptr<T>(
    constants: &DeviceMemory<T>,
    expected: usize,
) -> Result<*mut T> {
    validate_device_constant_len(constants.len(), expected)?;
    Ok(constants.as_ptr().cast_mut())
}