Skip to content

Commit 0102e25

Browse files
committed
Update
[ghstack-poisoned]
1 parent 3351d50 commit 0102e25

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

extension/llm/custom_ops/op_sdpa.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@
1919
#include <vector>
2020

2121
#ifdef ET_USE_THREADPOOL
22-
#include <executorch/runtime/kernel/thread_parallel_interface.h>
2322
#include <executorch/extension/threadpool/threadpool.h>
23+
#include <executorch/runtime/kernel/thread_parallel_interface.h>
2424
#endif
2525
#include <executorch/extension/kernel_util/make_boxed_from_unboxed_functor.h>
2626

extension/threadpool/test/thread_parallel_test.cpp

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,16 @@ class ParallelTest : public ::testing::TestWithParam<bool> {
4646
}
4747

4848
template <typename Func>
49-
bool parallel_for(const int64_t begin, const int64_t end, const int64_t grain_size, const Func& func) {
49+
bool parallel_for(
50+
const int64_t begin,
51+
const int64_t end,
52+
const int64_t grain_size,
53+
const Func& func) {
5054
if (GetParam()) {
5155
return executorch::extension::parallel_for(begin, end, grain_size, func);
5256
}
53-
return executorch::extension::internal::parallel_for_no_threadpool(begin, end, grain_size, func);
57+
return executorch::extension::internal::parallel_for_no_threadpool(
58+
begin, end, grain_size, func);
5459
}
5560

5661
std::array<int, 10> data_;
@@ -203,4 +208,7 @@ TEST_P(ParallelTest, TestChunkSizeTooLarge) {
203208
}
204209
}
205210

206-
INSTANTIATE_TEST_SUITE_P(ParallelTestWithOrWithoutThreadpool, ParallelTest, ::testing::Values(true, false));
211+
INSTANTIATE_TEST_SUITE_P(
212+
ParallelTestWithOrWithoutThreadpool,
213+
ParallelTest,
214+
::testing::Values(true, false));

0 commit comments

Comments
 (0)