cache stride for unchacked getters

TudbuT · May 11, 2023 · 3932da7 · 3932da7
1 parent 8b95cc0
commit 3932da7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "micro_ndarray"
-version = "0.2.4"
+version = "0.2.5"
 edition = "2021"
 repository = "https://github.com/tudbut/micro_ndarray"
 license = "MIT"

diff --git a/src/array.rs b/src/array.rs
@@ -127,9 +127,7 @@ impl<'a, T, const D: usize> Array<T, D> {
                 real_loc += dim;
                 continue;
             }
-            for s in &self.size[0..i] {
-                dim *= s;
-            }
+            dim *= self.stride[i];
             real_loc += dim;
         }
         self.data.get_unchecked(real_loc)
@@ -177,9 +175,7 @@ impl<'a, T, const D: usize> Array<T, D> {
                 real_loc += dim;
                 continue;
             }
-            for s in &self.size[0..i] {
-                dim *= s;
-            }
+            dim *= self.stride[i];
             real_loc += dim;
         }
         self.data.get_unchecked_mut(real_loc)