cubecl-core 0.10.0-pre.3

CubeCL core create
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

use crate::{self as cubecl};
use cubecl::prelude::*;

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

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

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

pub struct Identity;

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

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

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