From c40189a74e9eea7b0e47262aa36047590f7b1537 Mon Sep 17 00:00:00 2001
From: Scott Wolchok <swolchok@meta.com>
Date: Sun, 19 Oct 2025 17:38:32 -0700
Subject: [PATCH] Update

[ghstack-poisoned]
---
 extension/threadpool/threadpool.cpp | 2 +-
 extension/threadpool/threadpool.h   | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/extension/threadpool/threadpool.cpp b/extension/threadpool/threadpool.cpp
index f4d88e668d6..e244954de6e 100644
--- a/extension/threadpool/threadpool.cpp
+++ b/extension/threadpool/threadpool.cpp
@@ -76,7 +76,7 @@ bool ThreadPool::_unsafe_reset_threadpool(uint32_t new_thread_count) {
 }
 
 void ThreadPool::run(
-    const std::function<void(size_t)>& fn,
+    runtime::FunctionRef<void(size_t)> fn,
     const size_t range) {
   // Run on same thread if NoThreadPoolGuard guard is enabled
   if (NoThreadPoolGuard::is_enabled()) {
diff --git a/extension/threadpool/threadpool.h b/extension/threadpool/threadpool.h
index 16acad6e5fa..b4fe7e4be96 100644
--- a/extension/threadpool/threadpool.h
+++ b/extension/threadpool/threadpool.h
@@ -14,6 +14,8 @@
 
 #include <pthreadpool.h>
 
+#include <executorch/runtime/core/function_ref.h>
+
 /*
  * Threadpool Options:
  *
@@ -74,7 +76,7 @@ class ThreadPool final {
    * multiple threads with the scope of the guard When NoThreadPoolGuard is not
    * used all calls to run method are serialized.
    */
-  void run(const std::function<void(size_t)>& fn, size_t range);
+  void run(runtime::FunctionRef<void(size_t)> fn, size_t range);
 
  private:
   friend pthreadpool_t get_pthreadpool();