rstsr-core 0.7.0

An n-Dimension Rust Tensor Toolkit
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_dev::*;

#[duplicate_item(
     OpAPI               Op           ;
    [OpAddAssignAPI   ] [AddAssign   ];
    [OpSubAssignAPI   ] [SubAssign   ];
    [OpMulAssignAPI   ] [MulAssign   ];
    [OpDivAssignAPI   ] [DivAssign   ];
    [OpRemAssignAPI   ] [RemAssign   ];
    [OpBitOrAssignAPI ] [BitOrAssign ];
    [OpBitAndAssignAPI] [BitAndAssign];
    [OpBitXorAssignAPI] [BitXorAssign];
    [OpShlAssignAPI   ] [ShlAssign   ];
    [OpShrAssignAPI   ] [ShrAssign   ];
)]
pub trait OpAPI<TA, TB, D>
where
    D: DimAPI,
    Self: DeviceAPI<TA> + DeviceAPI<TB>,
{
    fn op_muta_refb(
        &self,
        a: &mut <Self as DeviceRawAPI<TA>>::Raw,
        la: &Layout<D>,
        b: &<Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
    ) -> Result<()>;

    fn op_muta_numb(&self, a: &mut <Self as DeviceRawAPI<TA>>::Raw, la: &Layout<D>, b: TB) -> Result<()>;
}

#[duplicate_item(
     OpAPI                 Op     ;
    [OpLConsumeAddAPI   ] [Add   ];
    [OpLConsumeSubAPI   ] [Sub   ];
    [OpLConsumeMulAPI   ] [Mul   ];
    [OpLConsumeDivAPI   ] [Div   ];
    [OpLConsumeRemAPI   ] [Rem   ];
    [OpLConsumeBitOrAPI ] [BitOr ];
    [OpLConsumeBitAndAPI] [BitAnd];
    [OpLConsumeBitXorAPI] [BitXor];
    [OpLConsumeShlAPI   ] [Shl   ];
    [OpLConsumeShrAPI   ] [Shr   ];
)]
pub trait OpAPI<TA, TB, D>
where
    D: DimAPI,
    Self: DeviceAPI<TA> + DeviceAPI<TB>,
{
    fn op_muta_refb(
        &self,
        a: &mut <Self as DeviceRawAPI<TA>>::Raw,
        la: &Layout<D>,
        b: &<Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
    ) -> Result<()>;

    fn op_muta_numb(&self, a: &mut <Self as DeviceRawAPI<TA>>::Raw, la: &Layout<D>, b: TB) -> Result<()>;
}

#[duplicate_item(
     OpAPI                 Op     ;
    [OpRConsumeAddAPI   ] [Add   ];
    [OpRConsumeSubAPI   ] [Sub   ];
    [OpRConsumeMulAPI   ] [Mul   ];
    [OpRConsumeDivAPI   ] [Div   ];
    [OpRConsumeRemAPI   ] [Rem   ];
    [OpRConsumeBitOrAPI ] [BitOr ];
    [OpRConsumeBitAndAPI] [BitAnd];
    [OpRConsumeBitXorAPI] [BitXor];
    [OpRConsumeShlAPI   ] [Shl   ];
    [OpRConsumeShrAPI   ] [Shr   ];
)]
pub trait OpAPI<TA, TB, D>
where
    D: DimAPI,
    Self: DeviceAPI<TA> + DeviceAPI<TB>,
{
    fn op_muta_refb(
        &self,
        b: &mut <Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
        a: &<Self as DeviceRawAPI<TA>>::Raw,
        la: &Layout<D>,
    ) -> Result<()>;

    fn op_muta_numb(&self, b: &mut <Self as DeviceRawAPI<TB>>::Raw, lb: &Layout<D>, a: TA) -> Result<()>;
}

#[duplicate_item(
     OpAPI      Op  ;
    [OpNegAPI] [Neg];
    [OpNotAPI] [Not];
)]
pub trait OpAPI<TA, TB, D>
where
    D: DimAPI,
    Self: DeviceAPI<MaybeUninit<TA>> + DeviceAPI<TA> + DeviceAPI<TB>,
{
    fn op_muta_refb(
        &self,
        a: &mut <Self as DeviceRawAPI<MaybeUninit<TA>>>::Raw,
        la: &Layout<D>,
        b: &<Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
    ) -> Result<()>;

    fn op_muta(&self, a: &mut <Self as DeviceRawAPI<TA>>::Raw, la: &Layout<D>) -> Result<()>;
}