custos 0.7.0

A minimal OpenCL, WGPU, CUDA and host CPU array manipulation engine.
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use crate::{prelude::Float, Combiner, Eval};

#[cfg(not(feature = "no-std"))]
use super::ToCLSource;

pub struct Exp<C> {
    pub comb: C,
}

impl<C> Combiner for Exp<C> {}

impl<T: Float, C: Eval<T>> Eval<T> for Exp<C> {
    #[inline]
    fn eval(self) -> T {
        self.comb.eval().exp()
    }
}

#[cfg(not(feature = "no-std"))]
impl<C: ToCLSource> ToCLSource for Exp<C> {
    #[inline]
    fn to_cl_source(&self) -> String {
        format!("exp({})", self.comb.to_cl_source())
    }
}

pub struct Sin<C> {
    pub comb: C,
}

impl<C> Combiner for Sin<C> {}

impl<T: Float, C: Eval<T>> Eval<T> for Sin<C> {
    #[inline]
    fn eval(self) -> T {
        self.comb.eval().sin()
    }
}

#[cfg(not(feature = "no-std"))]
impl<C: ToCLSource> ToCLSource for Sin<C> {
    #[inline]
    fn to_cl_source(&self) -> String {
        format!("sin({})", self.comb.to_cl_source())
    }
}

pub struct Cos<C> {
    pub comb: C,
}

impl<C> Combiner for Cos<C> {}

impl<T: Float, C: Eval<T>> Eval<T> for Cos<C> {
    #[inline]
    fn eval(self) -> T {
        self.comb.eval().cos()
    }
}

#[cfg(not(feature = "no-std"))]
impl<C: ToCLSource> ToCLSource for Cos<C> {
    #[inline]
    fn to_cl_source(&self) -> String {
        format!("cos({})", self.comb.to_cl_source())
    }
}

pub struct Tan<C> {
    pub comb: C,
}

impl<C> Combiner for Tan<C> {}

impl<T: Float, C: Eval<T>> Eval<T> for Tan<C> {
    #[inline]
    fn eval(self) -> T {
        self.comb.eval().tan()
    }
}

#[cfg(not(feature = "no-std"))]
impl<C: ToCLSource> ToCLSource for Tan<C> {
    #[inline]
    fn to_cl_source(&self) -> String {
        format!("tan({})", self.comb.to_cl_source())
    }
}

pub struct Neg<C> {
    pub comb: C,
}

impl<C> Combiner for Neg<C> {}

impl<T: core::ops::Neg<Output = T>, C: Eval<T>> Eval<T> for Neg<C> {
    #[inline]
    fn eval(self) -> T {
        self.comb.eval().neg()
    }
}

#[cfg(not(feature = "no-std"))]
impl<C: ToCLSource> ToCLSource for Neg<C> {
    #[inline]
    fn to_cl_source(&self) -> String {
        format!("-({})", self.comb.to_cl_source())
    }
}