From 4251a777ec24b2b24248d9bb1b124bae56b6a055 Mon Sep 17 00:00:00 2001
From: Thiago Machado <swfsql@gmail.com>
Date: Tue, 5 Dec 2023 08:30:19 -0500
Subject: [PATCH] miri pass nn::layers::add_into::tests::longer_network

Thread main finished before other threads were still active. The only
thing related to threads were the gemm usage of rayon, and disabling
rayon usage for matmul seems to be sufficient to fix this.
---
 dfdx-core/src/tensor_ops/matmul/cpu_kernel.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/dfdx-core/src/tensor_ops/matmul/cpu_kernel.rs b/dfdx-core/src/tensor_ops/matmul/cpu_kernel.rs
index bf3e6ce0..e22af5a0 100644
--- a/dfdx-core/src/tensor_ops/matmul/cpu_kernel.rs
+++ b/dfdx-core/src/tensor_ops/matmul/cpu_kernel.rs
@@ -90,7 +90,7 @@ impl MatMulImpl<crate::dtypes::AMP<half::f16>> for Cpu {
                 false,
                 false,
                 false,
-                gemm::Parallelism::Rayon(rayon::current_num_threads()),
+                gemm::Parallelism::None,
             )
         }
     }
@@ -138,7 +138,7 @@ impl MatMulImpl<half::f16> for Cpu {
                 false,
                 false,
                 false,
-                gemm::Parallelism::Rayon(rayon::current_num_threads()),
+                gemm::Parallelism::None,
             )
         }
     }
@@ -180,7 +180,7 @@ impl MatMulImpl<f32> for Cpu {
                 false,
                 false,
                 false,
-                gemm::Parallelism::Rayon(rayon::current_num_threads()),
+                gemm::Parallelism::None,
             )
         }
     }
@@ -222,7 +222,7 @@ impl MatMulImpl<f64> for Cpu {
                 false,
                 false,
                 false,
-                gemm::Parallelism::Rayon(rayon::current_num_threads()),
+                gemm::Parallelism::None,
             )
         }
     }