rstsr_openblas/rayon_auto_impl/
adv_indexing.rs

1use crate::prelude_dev::*;
2
3impl<T, D> DeviceIndexSelectAPI<T, D> for DeviceRayonAutoImpl
4where
5    T: Clone + Send + Sync,
6    D: DimAPI + DimSmallerOneAPI,
7    D::SmallerOne: DimAPI,
8{
9    fn index_select(
10        &self,
11        c: &mut Vec<MaybeUninit<T>>,
12        lc: &Layout<D>,
13        a: &Vec<T>,
14        la: &Layout<D>,
15        axis: usize,
16        indices: &[usize],
17    ) -> Result<()> {
18        let pool = self.get_current_pool();
19        index_select_cpu_rayon(c, lc, a, la, axis, indices, pool)
20    }
21}