parallel_invoke.h

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */
 
#pragma once
 
#include <utility>
 
#include <dispenso/platform.h>
#include <dispenso/task_set.h>
 
namespace dispenso {
 
template <typename F>
DISPENSO_INLINE void parallel_invoke(ConcurrentTaskSet& /*tasks*/, F&& f) {
  // Single-functor base case: run inline.
  std::forward<F>(f)();
}
 
template <typename F1, typename F2, typename... Fs>
DISPENSO_INLINE void parallel_invoke(ConcurrentTaskSet& tasks, F1&& f1, F2&& f2, Fs&&... fs) {
  // skipRecheck=true: the per-call inline gate's TaskSet check is enough; the
  // pool-level recheck is redundant when the caller is in a fork-join
  // pattern.
  tasks.schedule(std::forward<F1>(f1), /*skipRecheck=*/true);
  parallel_invoke(tasks, std::forward<F2>(f2), std::forward<Fs>(fs)...);
}
 
} // namespace dispenso