singe-kernel 0.1.0-alpha.4

Reusable CPU and GPU kernels.
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

#![allow(dead_code)]

use singe_cuda::view::{DeviceRepr, DeviceSlice};

use crate::{error::Result, utility::ensure_len};

pub fn ensure_binary_lengths<T: DeviceRepr>(
    out: &impl DeviceSlice<T>,
    lhs: &impl DeviceSlice<T>,
    rhs: &impl DeviceSlice<T>,
) -> Result<usize> {
    let len = out.len();
    ensure_len(lhs.len(), len)?;
    ensure_len(rhs.len(), len)?;
    Ok(len)
}

pub fn ensure_unary_lengths<T: DeviceRepr>(
    out: &impl DeviceSlice<T>,
    input: &impl DeviceSlice<T>,
) -> Result<usize> {
    let len = out.len();
    ensure_len(input.len(), len)?;
    Ok(len)
}