flower/external/include/taskflow/algorithm/launch.hpp

#pragma once

#include "../core/async.hpp"

namespace tf {

// Function: launch_loop
template <typename P, typename Loop>
TF_FORCE_INLINE void launch_loop(
  size_t N, 
  size_t W, 
  Runtime& rt, 
  std::atomic<size_t>& next, 
  P&& part, 
  Loop&& loop
) {

  //static_assert(std::is_lvalue_reference_v<Loop>, "");
  
  using namespace std::string_literals;

  for(size_t w=0; w<W; w++) {
    auto r = N - next.load(std::memory_order_relaxed);
    // no more loop work to do - finished by previous async tasks
    if(!r) {
      break;
    }
    // tail optimization
    if(r <= part.chunk_size() || w == W-1) {
      loop();
      break;
    }
    else {
      rt.silent_async_unchecked("loop-"s + std::to_string(w), loop);
    }
  }
      
  rt.join();
}

// Function: launch_loop
template <typename Loop>
TF_FORCE_INLINE void launch_loop(
  size_t W,
  size_t w,
  Runtime& rt, 
  Loop&& loop 
) {
  using namespace std::string_literals;
  if(w == W-1) {
    loop();
  }
  else {
    rt.silent_async_unchecked("loop-"s + std::to_string(w), loop);
  }
}

}  // end of namespace tf -----------------------------------------------------