pipeline.h

Go to the documentation of this file.

/*
 * 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 <limits>
 
#include <dispenso/detail/pipeline_impl.h>
 
namespace dispenso {
 
template <typename T>
using OpResult = detail::OpResult<T>;
 
constexpr ssize_t kStageNoLimit = std::numeric_limits<ssize_t>::max();
 
template <typename F>
auto stage(F&& f, ssize_t limit) {
  return detail::Stage<F>(std::forward<F>(f), limit);
}
 
template <typename... Stages>
void pipeline(ThreadPool& pool, Stages&&... sIn) {
  ConcurrentTaskSet tasks(pool);
  auto pipes = detail::makePipes(tasks, std::forward<Stages>(sIn)...);
  pipes.execute();
  pipes.wait();
}
 
template <typename... Stages>
void pipeline(Stages&&... sIn) {
  pipeline(globalThreadPool(), std::forward<Stages>(sIn)...);
}
 
} // namespace dispenso