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

use crate::prelude_dev::*;

impl<TC, TA, DC, DA> OpAssignArbitaryAPI<TC, DC, DA, TA> for DeviceRayonAutoImpl
where
    TC: Clone + Send + Sync,
    TA: Clone + Send + Sync + DTypeCastAPI<TC>,
    DC: DimAPI,
    DA: DimAPI,
{
    fn assign_arbitary(&self, c: &mut Vec<TC>, lc: &Layout<DC>, a: &Vec<TA>, la: &Layout<DA>) -> Result<()> {
        let pool = self.get_current_pool();
        let default_order = self.default_order();
        assign_arbitary_promote_cpu_rayon(c, lc, a, la, default_order, pool)
    }

    fn assign_arbitary_uninit(
        &self,
        c: &mut Vec<MaybeUninit<TC>>,
        lc: &Layout<DC>,
        a: &Vec<TA>,
        la: &Layout<DA>,
    ) -> Result<()> {
        let pool = self.get_current_pool();
        let default_order = self.default_order();
        return assign_arbitary_uninit_promote_cpu_rayon(c, lc, a, la, default_order, pool);
    }
}

impl<TC, TA, D> OpAssignAPI<TC, D, TA> for DeviceRayonAutoImpl
where
    TC: Clone + Send + Sync,
    TA: Clone + Send + Sync + DTypeCastAPI<TC>,
    D: DimAPI,
{
    fn assign(&self, c: &mut Vec<TC>, lc: &Layout<D>, a: &Vec<TA>, la: &Layout<D>) -> Result<()> {
        let pool = self.get_current_pool();
        assign_promote_cpu_rayon(c, lc, a, la, pool)
    }

    fn assign_uninit(&self, c: &mut Vec<MaybeUninit<TC>>, lc: &Layout<D>, a: &Vec<TA>, la: &Layout<D>) -> Result<()> {
        let pool = self.get_current_pool();
        return assign_uninit_promote_cpu_rayon(c, lc, a, la, pool);
    }

    fn fill(&self, c: &mut Vec<TC>, lc: &Layout<D>, fill: TA) -> Result<()> {
        let pool = self.get_current_pool();
        fill_promote_cpu_rayon(c, lc, fill, pool)
    }
}