cubecl-core 0.12.0-pre.2

CubeCL core create
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::prelude::*;

/// An iterator item that can be read as a number, so `sum` and `product` accept both `T` and `&T`
/// like [`Sum`](core::iter::Sum) and [`Product`](core::iter::Product).
pub trait IntoNumericExpand<S: Numeric> {
    /// Read the item.
    fn into_numeric_expand(self, scope: &Scope) -> NativeExpand<S>;
}

impl<S: Numeric> IntoNumericExpand<S> for NativeExpand<S> {
    fn into_numeric_expand(self, _scope: &Scope) -> NativeExpand<S> {
        self
    }
}

impl<S: Numeric> IntoNumericExpand<S> for &NativeExpand<S> {
    fn into_numeric_expand(self, scope: &Scope) -> NativeExpand<S> {
        self.__expand_deref_method(scope)
    }
}