Merge pull request #934 from jturner314/miri-test-size

Reduce test size for miri
rust-ndarray · Mar 10, 2021 · 5bd5891 · 5bd5891
2 parents 61acbd3 + 784a9a8
commit 5bd5891
Showing 1 changed file with 15 additions and 11 deletions.
diff --git a/tests/oper.rs b/tests/oper.rs
@@ -666,17 +666,21 @@ fn gemm_64_1_f() {
 fn gen_mat_mul_i32() {
     let alpha = -1;
     let beta = 2;
-    let sizes = vec![
-        (4, 4, 4),
-        (8, 8, 8),
-        (17, 15, 16),
-        (4, 17, 3),
-        (17, 3, 22),
-        (19, 18, 2),
-        (16, 17, 15),
-        (15, 16, 17),
-        (67, 63, 62),
-    ];
+    let sizes = if cfg!(miri) {
+        vec![(4, 4, 4), (4, 7, 3)]
+    } else {
+        vec![
+            (4, 4, 4),
+            (8, 8, 8),
+            (17, 15, 16),
+            (4, 17, 3),
+            (17, 3, 22),
+            (19, 18, 2),
+            (16, 17, 15),
+            (15, 16, 17),
+            (67, 63, 62),
+        ]
+    };
     for &(m, k, n) in &sizes {
         let a = range_i32(m, k);
         let b = range_i32(k, n);