ruda-kernel 0.1.4

Ruda Rust kernel DSL and device tensor operations.
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

use crate::dsl::prelude::*;

#[ruda]
pub(crate) trait UnaryOp: 'static + Send + Sync {
    type Options: LaunchArg;

    fn do_stuff<C: RudaPrimitive>(input: C, option: Self::Options) -> C;
}

#[ruda(launch)]
pub(crate) fn associated_type_input<O: UnaryOp>(_options: &O::Options) {}

pub struct Identity;

#[ruda]
impl UnaryOp for Identity {
    type Options = ();

    fn do_stuff<C: RudaPrimitive>(input: C, _option: Self::Options) -> C {
        input
    }
}

#[ruda]
pub(crate) fn trait_as<C: RudaPrimitive>(val: C) -> C {
    <Identity as UnaryOp>::do_stuff::<C>(val, ())
}