ha-ndarray 0.5.0

A hardware-accelerated n-dimensional array
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

use memoize::memoize;
use ocl::Program;

use crate::Error;

use super::build;

#[memoize]
pub fn gather_cond(c_type: &'static str) -> Result<Program, Error> {
    let src = format!(
        r#"
        __kernel void gather_cond(
            __global const uchar* restrict cond,
            __global const {c_type}* restrict then,
            __global const {c_type}* restrict or_else,
            __global {c_type}* restrict output)
        {{
            const ulong offset = get_global_id(0);

            if (cond[offset] != 0) {{
                output[offset] = then[offset];
            }} else {{
                output[offset] = or_else[offset];
            }}
        }}
        "#,
    );

    build(&src)
}