revert some things and add poc for mut and immut iterators
This commit is contained in:
parent
9a3bebc5c0
commit
0fcd9dd702
|
@ -101,7 +101,7 @@ glam020 = { package = "glam", version = "0.20", optional = true }
|
||||||
glam021 = { package = "glam", version = "0.21", optional = true }
|
glam021 = { package = "glam", version = "0.21", optional = true }
|
||||||
glam022 = { package = "glam", version = "0.22", optional = true }
|
glam022 = { package = "glam", version = "0.22", optional = true }
|
||||||
cust_core = { version = "0.1", optional = true }
|
cust_core = { version = "0.1", optional = true }
|
||||||
|
rayon = "1.5" # TODO make this feature gated
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
|
|
165
src/base/iter.rs
165
src/base/iter.rs
|
@ -1,12 +1,17 @@
|
||||||
//! Matrix iterators.
|
//! Matrix iterators.
|
||||||
|
|
||||||
|
use core::fmt::Debug;
|
||||||
|
use core::ops::Range;
|
||||||
use std::iter::FusedIterator;
|
use std::iter::FusedIterator;
|
||||||
use std::marker::PhantomData;
|
use std::marker::PhantomData;
|
||||||
use std::mem;
|
use std::mem;
|
||||||
|
|
||||||
|
use rayon::iter::plumbing::Producer;
|
||||||
|
|
||||||
use crate::base::dimension::{Dim, U1};
|
use crate::base::dimension::{Dim, U1};
|
||||||
use crate::base::storage::{RawStorage, RawStorageMut};
|
use crate::base::storage::{RawStorage, RawStorageMut};
|
||||||
use crate::base::{Matrix, MatrixView, MatrixViewMut, Scalar};
|
use crate::base::{Matrix, MatrixView, MatrixViewMut, Scalar};
|
||||||
|
use crate::{DMatrix, DimMax};
|
||||||
|
|
||||||
macro_rules! iterator {
|
macro_rules! iterator {
|
||||||
(struct $Name:ident for $Storage:ident.$ptr: ident -> $Ptr:ty, $Ref:ty, $SRef: ty) => {
|
(struct $Name:ident for $Storage:ident.$ptr: ident -> $Ptr:ty, $Ref:ty, $SRef: ty) => {
|
||||||
|
@ -288,7 +293,6 @@ impl<'a, T: Scalar, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> ExactSizeIte
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
*
|
|
||||||
* Column iterators.
|
* Column iterators.
|
||||||
*
|
*
|
||||||
*/
|
*/
|
||||||
|
@ -296,12 +300,25 @@ impl<'a, T: Scalar, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> ExactSizeIte
|
||||||
/// An iterator through the columns of a matrix.
|
/// An iterator through the columns of a matrix.
|
||||||
pub struct ColumnIter<'a, T, R: Dim, C: Dim, S: RawStorage<T, R, C>> {
|
pub struct ColumnIter<'a, T, R: Dim, C: Dim, S: RawStorage<T, R, C>> {
|
||||||
mat: &'a Matrix<T, R, C, S>,
|
mat: &'a Matrix<T, R, C, S>,
|
||||||
curr: usize,
|
range: Range<usize>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorage<T, R, C>> ColumnIter<'a, T, R, C, S> {
|
impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorage<T, R, C>> ColumnIter<'a, T, R, C, S> {
|
||||||
|
/// a new column iterator covering all columns of the matrix
|
||||||
pub(crate) fn new(mat: &'a Matrix<T, R, C, S>) -> Self {
|
pub(crate) fn new(mat: &'a Matrix<T, R, C, S>) -> Self {
|
||||||
ColumnIter { mat, curr: 0 }
|
ColumnIter {
|
||||||
|
mat,
|
||||||
|
range: 0..mat.ncols(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// a new column iterator covering column indices [begin,end)
|
||||||
|
/// where begin is included in the range but index end is not
|
||||||
|
/// begin must lie in [0,ncols] and end must lie in [0,ncols].
|
||||||
|
pub(crate) fn with_range(mat: &'a Matrix<T, R, C, S>, range: Range<usize>) -> Self {
|
||||||
|
debug_assert!(range.end <= mat.ncols());
|
||||||
|
debug_assert!(range.start < mat.ncols());
|
||||||
|
debug_assert!(range.start <= range.end);
|
||||||
|
Self { mat, range }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -310,9 +327,10 @@ impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorage<T, R, C>> Iterator for ColumnIter
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
if self.curr < self.mat.ncols() {
|
debug_assert!(self.range.start <= self.range.end);
|
||||||
let res = self.mat.column(self.curr);
|
if self.range.start < self.range.end {
|
||||||
self.curr += 1;
|
let res = self.mat.column(self.range.start);
|
||||||
|
self.range.start += 1;
|
||||||
Some(res)
|
Some(res)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
|
@ -321,15 +339,29 @@ impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorage<T, R, C>> Iterator for ColumnIter
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||||
(
|
let hint = self.range.len();
|
||||||
self.mat.ncols() - self.curr,
|
(hint, Some(hint))
|
||||||
Some(self.mat.ncols() - self.curr),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn count(self) -> usize {
|
fn count(self) -> usize {
|
||||||
self.mat.ncols() - self.curr
|
self.range.len()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorage<T, R, C>> DoubleEndedIterator
|
||||||
|
for ColumnIter<'a, T, R, C, S>
|
||||||
|
{
|
||||||
|
fn next_back(&mut self) -> Option<Self::Item> {
|
||||||
|
debug_assert!(self.range.start <= self.range.end);
|
||||||
|
if !self.range.is_empty() {
|
||||||
|
self.range.end -= 1;
|
||||||
|
debug_assert!(self.range.end < self.mat.ncols());
|
||||||
|
debug_assert!(self.range.end >= self.range.start);
|
||||||
|
Some(self.mat.column(self.range.end))
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -338,29 +370,53 @@ impl<'a, T: Scalar, R: Dim, C: Dim, S: 'a + RawStorage<T, R, C>> ExactSizeIterat
|
||||||
{
|
{
|
||||||
#[inline]
|
#[inline]
|
||||||
fn len(&self) -> usize {
|
fn len(&self) -> usize {
|
||||||
self.mat.ncols() - self.curr
|
self.range.end - self.range.start
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols>> Producer for ColumnIter<'a, T, R, Cols, S>
|
||||||
|
where
|
||||||
|
T: Send + Sync + Debug + PartialEq + Clone + 'static,
|
||||||
|
S: Sync,
|
||||||
|
{
|
||||||
|
type Item = MatrixSlice<'a, T, R, U1, S::RStride, S::CStride>;
|
||||||
|
type IntoIter = ColumnIter<'a, T, R, Cols, S>;
|
||||||
|
|
||||||
|
fn split_at(self, index: usize) -> (Self, Self) {
|
||||||
|
// the index is relative to the size of this current iterator
|
||||||
|
// it will always start at zero
|
||||||
|
let left = Self {
|
||||||
|
mat: self.mat,
|
||||||
|
range: self.range.start..(self.range.start + index),
|
||||||
|
};
|
||||||
|
|
||||||
|
let right = Self {
|
||||||
|
mat: self.mat,
|
||||||
|
range: (self.range.start + index)..self.range.end,
|
||||||
|
};
|
||||||
|
(left, right)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn into_iter(self) -> Self::IntoIter {
|
||||||
|
self
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// An iterator through the mutable columns of a matrix.
|
/// An iterator through the mutable columns of a matrix.
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct ColumnIterMut<'a, T, R: Dim, C: Dim, S: RawStorageMut<T, R, C>> {
|
pub struct ColumnIterMut<'a, T, R: Dim, C: Dim, S: RawStorageMut<T, R, C>> {
|
||||||
mat: *mut Matrix<T, R, C, S>,
|
mat: &'a mut Matrix<T, R, C, S>,
|
||||||
curr: usize,
|
range: Range<usize>,
|
||||||
phantom: PhantomData<&'a mut Matrix<T, R, C, S>>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> ColumnIterMut<'a, T, R, C, S> {
|
impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> ColumnIterMut<'a, T, R, C, S> {
|
||||||
pub(crate) fn new(mat: &'a mut Matrix<T, R, C, S>) -> Self {
|
pub(crate) fn new(mat: &'a mut Matrix<T, R, C, S>) -> Self {
|
||||||
ColumnIterMut {
|
let range = 0..mat.ncols();
|
||||||
mat,
|
ColumnIterMut { mat, range }
|
||||||
curr: 0,
|
|
||||||
phantom: PhantomData,
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn ncols(&self) -> usize {
|
fn ncols(&self) -> usize {
|
||||||
unsafe { (*self.mat).ncols() }
|
self.mat.ncols()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -370,10 +426,11 @@ impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> Iterator
|
||||||
type Item = MatrixViewMut<'a, T, R, U1, S::RStride, S::CStride>;
|
type Item = MatrixViewMut<'a, T, R, U1, S::RStride, S::CStride>;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
fn next(&'_ mut self) -> Option<Self::Item> {
|
||||||
if self.curr < self.ncols() {
|
if self.range.start < self.ncols() {
|
||||||
let res = unsafe { (*self.mat).column_mut(self.curr) };
|
let pmat: *mut _ = self.mat;
|
||||||
self.curr += 1;
|
let res = unsafe { (*pmat).column_mut(self.range.start) };
|
||||||
|
self.range.start += 1;
|
||||||
Some(res)
|
Some(res)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
|
@ -382,12 +439,13 @@ impl<'a, T, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> Iterator
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||||
(self.ncols() - self.curr, Some(self.ncols() - self.curr))
|
let hint = self.range.len();
|
||||||
|
(hint, Some(hint))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn count(self) -> usize {
|
fn count(self) -> usize {
|
||||||
self.ncols() - self.curr
|
self.range.len()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -396,6 +454,57 @@ impl<'a, T: Scalar, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> ExactSizeIte
|
||||||
{
|
{
|
||||||
#[inline]
|
#[inline]
|
||||||
fn len(&self) -> usize {
|
fn len(&self) -> usize {
|
||||||
self.ncols() - self.curr
|
self.range.len()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<'a, T: Scalar, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> DoubleEndedIterator
|
||||||
|
for ColumnIterMut<'a, T, R, C, S>
|
||||||
|
{
|
||||||
|
fn next_back(&mut self) -> Option<Self::Item> {
|
||||||
|
debug_assert!(self.range.start <= self.range.end);
|
||||||
|
if !self.range.is_empty() {
|
||||||
|
self.range.end -= 1;
|
||||||
|
debug_assert!(self.range.end < unsafe { (*self.mat).ncols() });
|
||||||
|
debug_assert!(self.range.end >= self.range.start);
|
||||||
|
let pmat: *mut _ = self.mat;
|
||||||
|
Some(unsafe { (*pmat).column_mut(self.range.end) })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T: Scalar, R: Dim, C: Dim, S: 'a + RawStorageMut<T, R, C>> Producer for ColumnIterMut<'a,T,R,C,S>
|
||||||
|
where T : Send + Sync + Debug + PartialEq + Clone,
|
||||||
|
S: Send + Sync {
|
||||||
|
type Item = MatrixSliceMut<'a, T, R, U1, S::RStride, S::CStride>;
|
||||||
|
type IntoIter = ColumnIterMut<'a,T,R,C,S>;
|
||||||
|
|
||||||
|
fn into_iter(self) -> Self::IntoIter {
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
fn split_at(self, index: usize) -> (Self, Self) {
|
||||||
|
// the index is relative to the size of this current iterator
|
||||||
|
// it will always start at zero
|
||||||
|
let pmat : * mut _ = self.mat;
|
||||||
|
|
||||||
|
let left = Self {
|
||||||
|
mat: unsafe {&mut *pmat},
|
||||||
|
range: self.range.start..(self.range.start + index),
|
||||||
|
};
|
||||||
|
|
||||||
|
let right = Self {
|
||||||
|
mat: self.mat,
|
||||||
|
range: (self.range.start + index)..self.range.end,
|
||||||
|
};
|
||||||
|
(left, right)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_send<T: Send>(_: T) {}
|
||||||
|
|
||||||
|
fn something(mut matrix: DMatrix<f32>) {
|
||||||
|
test_send(matrix.column_iter_mut());
|
||||||
|
}
|
||||||
|
|
|
@ -0,0 +1,169 @@
|
||||||
|
//! this module implements parallelators to make matrices work with
|
||||||
|
//! the rayon crate seamlessly
|
||||||
|
|
||||||
|
use core::{
|
||||||
|
fmt::Debug,
|
||||||
|
iter::{Skip, Take},
|
||||||
|
marker::PhantomData,
|
||||||
|
ops::Range,
|
||||||
|
};
|
||||||
|
use std::os::unix::prelude::AsRawFd;
|
||||||
|
|
||||||
|
use rayon::{
|
||||||
|
iter::plumbing::{bridge, Producer},
|
||||||
|
prelude::*,
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
iter::{ColumnIter, ColumnIterMut}, Const, DMatrix, Dim, Dynamic, Matrix, MatrixSlice, MatrixSliceMut,
|
||||||
|
RawStorage, RawStorageMut, U1, SliceStorageMut,
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::conversion;
|
||||||
|
|
||||||
|
/// a rayon parallel iterator over the columns of a matrix
|
||||||
|
pub struct ParColumnIter<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols>> {
|
||||||
|
mat: &'a Matrix<T, R, Cols, S>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols>> ParColumnIter<'a, T, R, Cols, S> {
|
||||||
|
fn new(matrix: &'a Matrix<T, R, Cols, S>) -> Self {
|
||||||
|
Self { mat: matrix }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols>> ParallelIterator
|
||||||
|
for ParColumnIter<'a, T, R, Cols, S>
|
||||||
|
where
|
||||||
|
T: Sync + Send + Clone + Debug + PartialEq + 'static,
|
||||||
|
S: Sync,
|
||||||
|
{
|
||||||
|
type Item = MatrixSlice<'a, T, R, U1, S::RStride, S::CStride>;
|
||||||
|
|
||||||
|
fn drive_unindexed<Consumer>(self, consumer: Consumer) -> Consumer::Result
|
||||||
|
where
|
||||||
|
Consumer: rayon::iter::plumbing::UnindexedConsumer<Self::Item>,
|
||||||
|
{
|
||||||
|
bridge(self, consumer)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn opt_len(&self) -> Option<usize> {
|
||||||
|
Some(self.mat.ncols())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols>> IndexedParallelIterator
|
||||||
|
for ParColumnIter<'a, T, R, Cols, S>
|
||||||
|
where
|
||||||
|
T: Send + Sync + Clone + Debug + PartialEq + 'static,
|
||||||
|
S: Sync,
|
||||||
|
{
|
||||||
|
fn len(&self) -> usize {
|
||||||
|
self.mat.ncols()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn drive<C: rayon::iter::plumbing::Consumer<Self::Item>>(self, consumer: C) -> C::Result {
|
||||||
|
bridge(self, consumer)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn with_producer<CB: rayon::iter::plumbing::ProducerCallback<Self::Item>>(
|
||||||
|
self,
|
||||||
|
callback: CB,
|
||||||
|
) -> CB::Output {
|
||||||
|
let producer = ColumnIter::new(self.mat);
|
||||||
|
callback.callback(producer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols>> Matrix<T, R, Cols, S>
|
||||||
|
where
|
||||||
|
T: Send + Sync + Clone + Debug + PartialEq + 'static,
|
||||||
|
S: Sync,
|
||||||
|
{
|
||||||
|
fn par_column_iter(&self) -> ParColumnIter<'_, T, R, Cols, S> {
|
||||||
|
ParColumnIter::new(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// TODO
|
||||||
|
pub struct ParColumnIterMut<'a,T,R:Dim ,Cols:Dim, S:RawStorage<T,R,Cols>+RawStorageMut<T,R,Cols>> {
|
||||||
|
mat : &'a mut Matrix<T,R,Cols,S>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a,T,R,Cols,S> ParColumnIterMut<'a,T,R,Cols,S>
|
||||||
|
where R: Dim, Cols : Dim, S:RawStorage<T,R,Cols> + RawStorageMut<T,R,Cols> {
|
||||||
|
/// TODO
|
||||||
|
pub fn new(mat : &'a mut Matrix<T,R,Cols,S>) -> Self {
|
||||||
|
Self {
|
||||||
|
mat,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a,T,R,Cols,S> ParallelIterator for ParColumnIterMut<'a,T,R,Cols,S>
|
||||||
|
where R: Dim, Cols : Dim, S:RawStorage<T,R,Cols> + RawStorageMut<T,R,Cols>,
|
||||||
|
T : Send + Sync + Debug + PartialEq + Clone + 'static,
|
||||||
|
S : Send + Sync {
|
||||||
|
type Item = MatrixSliceMut<'a, T, R, U1, S::RStride, S::CStride>;
|
||||||
|
fn drive_unindexed<C>(self, consumer: C) -> C::Result
|
||||||
|
where
|
||||||
|
C: rayon::iter::plumbing::UnindexedConsumer<Self::Item> {
|
||||||
|
bridge(self,consumer)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn opt_len(&self) -> Option<usize> {
|
||||||
|
Some(self.mat.ncols())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
impl<'a,T,R,Cols,S> IndexedParallelIterator for ParColumnIterMut<'a,T,R,Cols,S>
|
||||||
|
where R: Dim, Cols : Dim, S:RawStorage<T,R,Cols> + RawStorageMut<T,R,Cols>,
|
||||||
|
T : Send + Sync + Debug + PartialEq + Clone + 'static,
|
||||||
|
S : Send + Sync {
|
||||||
|
fn drive<C: rayon::iter::plumbing::Consumer<Self::Item>>(self, consumer: C) -> C::Result {
|
||||||
|
bridge(self,consumer)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn len(&self) -> usize {
|
||||||
|
self.mat.ncols()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn with_producer<CB: rayon::iter::plumbing::ProducerCallback<Self::Item>>(self, callback: CB) -> CB::Output {
|
||||||
|
let producer = ColumnIterMut::new(self.mat);
|
||||||
|
callback.callback(producer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, T, R: Dim, Cols: Dim, S: RawStorage<T, R, Cols> + RawStorageMut<T,R,Cols>> Matrix<T, R, Cols, S>
|
||||||
|
where
|
||||||
|
T: Send + Sync + Clone + Debug + PartialEq + 'static,
|
||||||
|
S: Sync,
|
||||||
|
{
|
||||||
|
fn par_column_iter_mut(&mut self) -> ParColumnIterMut<'_, T, R, Cols, S> {
|
||||||
|
ParColumnIterMut::new(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn parallel_iterator() {
|
||||||
|
let matrix = DMatrix::<f32>::zeros(3, 4);
|
||||||
|
let res: Vec<_> = matrix.par_column_iter().map(|col| col.len()).collect();
|
||||||
|
assert_eq!(res, vec![3, 3, 3, 3]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_mut_parallel_iter() {
|
||||||
|
let mut matrix = DMatrix::<f32>::zeros(4, 3);
|
||||||
|
matrix.par_column_iter_mut().enumerate().for_each(|(idx,mut col)| col[idx]=1f32);
|
||||||
|
let identity = DMatrix::<f32>::identity(4, 3);
|
||||||
|
assert_eq!(matrix,identity);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
fn try_some_stuff() {
|
||||||
|
let mut mat = DMatrix::<f32>::zeros(3, 4);
|
||||||
|
let _left = mat.columns_mut(0, 1);
|
||||||
|
let _right = mat.columns_mut(1, 3);
|
||||||
|
}
|
Loading…
Reference in New Issue