diff --git a/gemma/ops_test.cc b/gemma/ops_test.cc index aec0d7f..9e0d4ec 100644 --- a/gemma/ops_test.cc +++ b/gemma/ops_test.cc @@ -531,7 +531,7 @@ void AssertClose(const MatT* HWY_RESTRICT expected, template void TestTiledMatMul() { - hwy::ThreadPool pool(3); + hwy::ThreadPool pool(hwy::ThreadPool::MaxThreads()); std::unique_ptr> a = GenerateMatHeap(0, pool); std::unique_ptr> b = @@ -570,7 +570,7 @@ void TestAllTiledMatMul() { } void TestMatVecAdd() { - hwy::ThreadPool pool(0); + hwy::ThreadPool pool(hwy::ThreadPool::MaxThreads()); constexpr size_t kOuter = 128 * 3; constexpr size_t kInner = 128 * 5; CompressedArray mat = @@ -590,7 +590,7 @@ void TestMatVecAdd() { } void TestTwoMatVecAdd() { - hwy::ThreadPool pool(0); + hwy::ThreadPool pool(hwy::ThreadPool::MaxThreads()); constexpr size_t kOuter = 128 * 3; constexpr size_t kInner = 128 * 5; CompressedArray mat0 = @@ -617,7 +617,7 @@ void TestTwoMatVecAdd() { } void TestTwoOfsMatVecAddLoop() { - hwy::ThreadPool pool(0); + hwy::ThreadPool pool(hwy::ThreadPool::MaxThreads()); constexpr size_t kOuter = 128 * 3; constexpr size_t kInner = 128 * 5; CompressedArray mat =