update

bokutotu · Sep 23, 2024 · 32f347f · 32f347f
1 parent 10e02ed
commit 32f347f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/zenu-matrix/src/operation/max.rs b/zenu-matrix/src/operation/max.rs
@@ -43,6 +43,9 @@ impl MaxIdx for Nvidia {
 impl<T: Num, R: Repr<Item = T>, D: Device> Matrix<R, DimDyn, D> {
     #[must_use]
     pub fn max_idx(&self) -> DimDyn {
+        if self.shape().is_empty() {
+            return DimDyn::from(&[] as &[usize]);
+        }
         let default_stride = self.to_default_stride();
         let idx = <D as MaxIdx>::max_idx(
             default_stride.as_ptr(),

diff --git a/zenu-test/src/lib.rs b/zenu-test/src/lib.rs
@@ -8,15 +8,15 @@ macro_rules! assert_mat_eq_epsilon {
         let epsilon = $epsilon;
         let diff = mat.to_ref() - mat2.to_ref();
         let abs = diff.abs();
-        let diff_asum = abs.asum();
-        if diff_asum > epsilon {
+        let diff_max = abs.max_item();
+        if diff_max > epsilon {
             panic!(
                 "assertion failed: `(left == right)`\n\
                 left: \n{:?},\n\
                 right: \n{:?}\n\
                 diff: \n{:?}\n\
-                diff_asum: \n{:?}",
-                mat, mat2, diff, diff_asum
+                diff_max: \n{:?}",
+                mat, mat2, diff, diff_max
             );
         }
     }};