Skip to content

Commit 8893683

Browse files
committed
fix indexer calls in <arith>Assign trait impls
1 parent 1c19542 commit 8893683

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

src/arith/mod.rs

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -462,10 +462,9 @@ macro_rules! arith_assign_func {
462462
fn $fn_name(&mut self, rhs: Array) {
463463
let tmp_seq = Seq::<f32>::default();
464464
let mut idxrs = Indexer::new();
465-
idxrs.set_index(&tmp_seq, 0, Some(false));
466-
idxrs.set_index(&tmp_seq, 1, Some(false));
467-
idxrs.set_index(&tmp_seq, 2, Some(false));
468-
idxrs.set_index(&tmp_seq, 3, Some(false));
465+
for n in 0..self.numdims() {
466+
idxrs.set_index(&tmp_seq, n, Some(false));
467+
}
469468
let tmp = assign_gen(self as &Array, &idxrs,
470469
& $func(self as &Array, &rhs, false));
471470
let old = mem::replace(self, tmp);
@@ -490,10 +489,9 @@ macro_rules! bit_assign_func {
490489
fn $fn_name(&mut self, rhs: Array) {
491490
let tmp_seq = Seq::<f32>::default();
492491
let mut idxrs = Indexer::new();
493-
idxrs.set_index(&tmp_seq, 0, Some(false));
494-
idxrs.set_index(&tmp_seq, 1, Some(false));
495-
idxrs.set_index(&tmp_seq, 2, Some(false));
496-
idxrs.set_index(&tmp_seq, 3, Some(false));
492+
for n in 0..self.numdims() {
493+
idxrs.set_index(&tmp_seq, n, Some(false));
494+
}
497495
let tmp = assign_gen(self as &Array, &idxrs, & $func(self as &Array, &rhs, false));
498496
let old = mem::replace(self, tmp);
499497
}

0 commit comments

Comments
 (0)