zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-15 21:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert changes in pocketfft.h

Browse Source
This commit is contained in:
Hanyu Deng 2025-04-10 22:37:36 +08:00 committed by Awni Hannun
parent b7ce3c947c
commit 27ceff33d3
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mlx/3rdparty/pocketfft.h vendored
View File

@ -537,11 +537,7 @@ inline size_t &num_threads()
static thread_local size_t num_threads_=1;
return num_threads_;
}
inline const size_t &max_threads()
{
static thread_local const size_t max_threads_ = std::max(1u, std::thread::hardware_concurrency());
return max_threads_;
}
static const size_t max_threads = std::max(1u, std::thread::hardware_concurrency());
class latch
{
@ -725,7 +721,7 @@ class thread_pool
workers_(nthreads)
{ create_threads(); }
thread_pool(): thread_pool(max_threads()) {}
thread_pool(): thread_pool(max_threads) {}
~thread_pool() { shutdown(); }
@ -790,7 +786,7 @@ template <typename Func>
void thread_map(size_t nthreads, Func f)
{
if (nthreads == 0)
nthreads = max_threads();
nthreads = max_threads;
if (nthreads == 1)
{ f(); return; }