make_tt.h

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-3-Clause
// to be #include'd within runtime::ttg namespace
 
#ifndef TTG_MAKE_TT_H
#define TTG_MAKE_TT_H
 
 
// Class to wrap a callable with signature
//
// case 1 (keyT != void): returnT op(auto&& key, input_valuesT&&..., std::tuple<output_terminalsT...>&)
// case 2 (keyT == void): returnT op(input_valuesT&&..., std::tuple<output_terminalsT...>&)
//
// returnT is void for funcT = synchronous (ordinary) function and the appropriate return type for funcT=coroutine
template <typename funcT, typename returnT, bool funcT_receives_input_tuple,
          bool funcT_receives_outterm_tuple, ttg::ExecutionSpace space,
          typename keyT, typename output_terminalsT, typename... input_valuesT>
class CallableWrapTT
    : public TT<
          keyT, output_terminalsT,
          CallableWrapTT<funcT, returnT, funcT_receives_input_tuple, funcT_receives_outterm_tuple, space, keyT, output_terminalsT, input_valuesT...>,
          ttg::typelist<input_valuesT...>, space> {
  using baseT = typename CallableWrapTT::ttT;
 
  using input_values_tuple_type = typename baseT::input_values_tuple_type;
  using input_refs_tuple_type = typename baseT::input_refs_tuple_type;
  using input_edges_type = typename baseT::input_edges_type;
  using output_edges_type = typename baseT::output_edges_type;
 
  using noref_funcT = std::remove_reference_t<funcT>;
  std::conditional_t<std::is_function_v<noref_funcT>, std::add_pointer_t<noref_funcT>, noref_funcT> func;
 
  using op_return_type =
#ifdef TTG_HAVE_COROUTINE
      std::conditional_t<std::is_same_v<returnT, ttg::resumable_task>,
                         ttg::coroutine_handle<ttg::resumable_task_state>,
#ifdef TTG_HAVE_DEVICE
                         std::conditional_t<std::is_same_v<returnT, ttg::device::Task>,
                                            ttg::device::Task::base_type,
                                            void>
#else  // TTG_HAVE_DEVICE
                           void
#endif  // TTG_HAVE_DEVICE
                         >;
#else   // TTG_HAVE_COROUTINE
      void;
#endif  // TTG_HAVE_COROUTINE
 
protected:
 
  template<typename ReturnT>
  auto process_return(ReturnT&& ret, output_terminalsT &out) {
    static_assert(std::is_same_v<std::remove_reference_t<decltype(ret)>, returnT>,
                  "CallableWrapTT<funcT,returnT,...>: returnT does not match the actual return type of funcT");
    if constexpr (!std::is_void_v<returnT>) {  // protect from compiling for void returnT
#ifdef TTG_HAVE_COROUTINE
      if constexpr (std::is_same_v<returnT, ttg::resumable_task>) {
        ttg::coroutine_handle<ttg::resumable_task_state> coro_handle;
        // if task completed destroy it
        if (ret.completed()) {
          ret.destroy();
        } else {  // if task is suspended return the coroutine promise ptr
          coro_handle = ret;
        }
        return coro_handle;
      } else
#ifdef TTG_HAVE_DEVICE
          if constexpr (std::is_same_v<returnT, ttg::device::Task>) {
        ttg::device::Task::base_type coro_handle = ret;
        return coro_handle;
      }
#else  // TTG_HAVE_DEVICE
        ttg::abort();  // should not happen
#endif  // TTG_HAVE_DEVICE
      if constexpr (!(std::is_same_v<returnT, ttg::resumable_task>
#ifdef TTG_HAVE_DEVICE
          || std::is_same_v<returnT, ttg::device::Task>
#endif  // TTG_HAVE_DEVICE
              ))
#endif
      {
        static_assert(std::tuple_size_v<std::remove_reference_t<decltype(out)>> == 1,
                      "CallableWrapTT<funcT,returnT,funcT_receives_outterm_tuple=true,...): funcT can return a "
                      "value only if there is only 1 out terminal");
        static_assert(std::tuple_size_v<returnT> <= 2,
                      "CallableWrapTT<funcT,returnT,funcT_receives_outterm_tuple=true,...): funcT can return a "
                      "value only if it is a plain value (then sent with null key), a tuple-like containing a single "
                      "key (hence value is void), or a tuple-like containing a key and a value");
        if constexpr (std::tuple_size_v<returnT> == 0)
          std::get<0>(out).sendv(std::move(ret));
        else if constexpr (std::tuple_size_v<returnT> == 1)
          std::get<0>(out).sendk(std::move(std::get<0>(ret)));
        else if constexpr (std::tuple_size_v<returnT> == 2)
          std::get<0>(out).send(std::move(std::get<0>(ret)), std::move(std::get<1>(ret)));
        return;
      }
    }
  }
 
  template <typename Key, typename Tuple, std::size_t... S>
  auto call_func(Key &&key, Tuple &&args_tuple, output_terminalsT &out, std::index_sequence<S...>) {
    using func_args_t = ttg::meta::tuple_concat_t<std::tuple<const Key &>, input_refs_tuple_type, output_edges_type>;
    auto invoke_func_handle_ret = [&](auto&&... args){
      if constexpr (std::is_void_v<returnT>) {
        func(std::forward<Key>(key), std::forward<decltype(args)>(args)...);
      } else {
        return process_return(func(std::forward<Key>(key), std::forward<decltype(args)>(args)...), out);
      }
    };
    auto unpack_input_tuple_if_needed = [&](auto&&... args){
      if constexpr (funcT_receives_input_tuple) {
        return invoke_func_handle_ret(std::forward<Tuple>(args_tuple), std::forward<decltype(args)>(args)...);
      } else {
        return invoke_func_handle_ret(baseT::template get<S, std::tuple_element_t<S + 1, func_args_t>>(std::forward<Tuple>(args_tuple))...,
                  std::forward<decltype(args)>(args)...);
      }
    };
 
    if constexpr (funcT_receives_outterm_tuple) {
      return unpack_input_tuple_if_needed(out);
    } else {
      auto old_output_tls_ptr = this->outputs_tls_ptr_accessor();
      this->set_outputs_tls_ptr();
      // make sure the output tls is reset
      auto _ = ttg::detail::scope_exit(
        [this, old_output_tls_ptr](){
          this->set_outputs_tls_ptr(old_output_tls_ptr);
        });
      return unpack_input_tuple_if_needed();
    }
  }
 
  template <typename Tuple, std::size_t... S>
  auto call_func(Tuple &&args_tuple, output_terminalsT &out, std::index_sequence<S...>) {
    using func_args_t = ttg::meta::tuple_concat_t<input_refs_tuple_type, output_edges_type>;
 
    auto invoke_func_handle_ret = [&](auto&&... args){
      if constexpr (std::is_void_v<returnT>) {
        func(std::forward<decltype(args)>(args)...);
      } else {
        return process_return(func(std::forward<decltype(args)>(args)...), out);
      }
    };
    auto unpack_input_tuple_if_needed = [&](auto&& fn, auto&&... args){
      if constexpr (funcT_receives_input_tuple) {
        return fn(std::forward<Tuple>(args_tuple), std::forward<decltype(args)>(args)...);
      } else {
        return fn(baseT::template get<S, std::tuple_element_t<S, func_args_t>>(std::forward<Tuple>(args_tuple))...,
                  std::forward<decltype(args)>(args)...);
      }
    };
 
    if constexpr (funcT_receives_outterm_tuple) {
      return unpack_input_tuple_if_needed(invoke_func_handle_ret, out);
    } else {
      auto old_output_tls_ptr = this->outputs_tls_ptr_accessor();
      this->set_outputs_tls_ptr();
      // make sure the output tls is reset
      auto _ = ttg::detail::scope_exit(
        [this, old_output_tls_ptr](){
          this->set_outputs_tls_ptr(old_output_tls_ptr);
        });
      return unpack_input_tuple_if_needed(invoke_func_handle_ret);
    }
  }
 
  template <typename Key>
  auto call_func(Key &&key, output_terminalsT &out) {
    auto invoke_func_handle_ret = [&](auto&&... args){
      if constexpr (std::is_void_v<returnT>) {
        func(std::forward<Key>(key), std::forward<decltype(args)>(args)...);
      } else {
        return process_return(func(std::forward<Key>(key), std::forward<decltype(args)>(args)...), out);
      }
    };
 
    auto invoke_func_empty_tuple = [&](auto&&... args){
      if constexpr(funcT_receives_input_tuple) {
        invoke_func_handle_ret(std::tuple<>{}, std::forward<decltype(args)>(args)...);
      } else {
        invoke_func_handle_ret(std::forward<decltype(args)>(args)...);
      }
    };
 
    if constexpr (funcT_receives_outterm_tuple) {
      invoke_func_handle_ret(out);
    } else {
      auto old_output_tls_ptr = this->outputs_tls_ptr_accessor();
      this->set_outputs_tls_ptr();
      // make sure the output tls is reset
      auto _ = ttg::detail::scope_exit(
        [this, old_output_tls_ptr](){
          this->set_outputs_tls_ptr(old_output_tls_ptr);
        });
      return invoke_func_handle_ret();
    }
  }
 
  template <typename OutputTerminals>
  auto call_func(OutputTerminals &out) {
 
    auto invoke_func_handle_ret = [&](auto&&... args){
      if constexpr (std::is_void_v<returnT>) {
        func(std::forward<decltype(args)>(args)...);
      } else {
        return process_return(func(std::forward<decltype(args)>(args)...), out);
      }
    };
 
    auto invoke_func_empty_tuple = [&](auto&&... args){
      if constexpr(funcT_receives_input_tuple) {
        return invoke_func_handle_ret(std::tuple<>{}, std::forward<decltype(args)>(args)...);
      } else {
        return invoke_func_handle_ret(std::forward<decltype(args)>(args)...);
      }
    };
 
    if constexpr (funcT_receives_outterm_tuple) {
      return invoke_func_empty_tuple(out);
    } else {
      auto old_output_tls_ptr = this->outputs_tls_ptr_accessor();
      this->set_outputs_tls_ptr();
      // make sure the output tls is reset
      auto _ = ttg::detail::scope_exit(
        [this, old_output_tls_ptr](){
          this->set_outputs_tls_ptr(old_output_tls_ptr);
        });
      return invoke_func_empty_tuple();
    }
  }
 
  template <typename Tuple, std::size_t... I>
  static auto make_output_terminal_ptrs(const Tuple &output_terminals, std::index_sequence<I...>) {
    return std::array<ttg::TerminalBase *, sizeof...(I)>{
        {static_cast<ttg::TerminalBase *>(&std::get<I>(output_terminals))...}};
  }
 
 public:
  template <typename funcT_>
  CallableWrapTT(funcT_ &&f, const input_edges_type &inedges, const typename baseT::output_edges_type &outedges,
                     const std::string &name, const std::vector<std::string> &innames,
                     const std::vector<std::string> &outnames)
      : baseT(inedges, outedges, name, innames, outnames), func(std::forward<funcT_>(f)) {}
 
  template <typename funcT_>
  CallableWrapTT(funcT_ &&f, const std::string &name, const std::vector<std::string> &innames,
                     const std::vector<std::string> &outnames)
      : baseT(name, innames, outnames), func(std::forward<funcT_>(f)) {}
 
  template <typename Key, typename ArgsTuple>
  std::enable_if_t<std::is_same_v<ArgsTuple, input_refs_tuple_type> &&
                       !ttg::meta::is_empty_tuple_v<input_refs_tuple_type> && !ttg::meta::is_void_v<Key>,
                   op_return_type>
  op(Key &&key, ArgsTuple &&args_tuple, output_terminalsT &out) {
    assert(&out == &baseT::get_output_terminals());
    return call_func(std::forward<Key>(key), std::forward<ArgsTuple>(args_tuple), out,
                     std::make_index_sequence<std::tuple_size_v<ArgsTuple>>{});
  };
 
  template <typename ArgsTuple, typename Key = keyT>
  std::enable_if_t<std::is_same_v<ArgsTuple, input_refs_tuple_type> &&
                       !ttg::meta::is_empty_tuple_v<input_refs_tuple_type> && ttg::meta::is_void_v<Key>,
                   op_return_type>
  op(ArgsTuple &&args_tuple, output_terminalsT &out) {
    assert(&out == &baseT::get_output_terminals());
    return call_func(std::forward<ArgsTuple>(args_tuple), out,
                     std::make_index_sequence<std::tuple_size_v<ArgsTuple>>{});
  };
 
  template <typename Key, typename ArgsTuple = input_refs_tuple_type>
  std::enable_if_t<ttg::meta::is_empty_tuple_v<ArgsTuple> && !ttg::meta::is_void_v<Key>, op_return_type> op(
      Key &&key, output_terminalsT &out) {
    assert(&out == &baseT::get_output_terminals());
    return call_func(std::forward<Key>(key), out);
  };
 
  template <typename Key = keyT, typename ArgsTuple = input_refs_tuple_type>
  std::enable_if_t<ttg::meta::is_empty_tuple_v<ArgsTuple> && ttg::meta::is_void_v<Key>, op_return_type> op(
      output_terminalsT &out) {
    assert(&out == &baseT::get_output_terminals());
    return call_func(out);
  };
};
 
template <typename funcT, typename returnT, bool funcT_receives_input_tuple,
          bool funcT_receives_outterm_tuple, ttg::ExecutionSpace space,
          typename keyT, typename output_terminalsT, typename input_values_typelistT>
struct CallableWrapTTAsTypelist;
 
template <typename funcT, typename returnT, bool funcT_receives_input_tuple,
          bool funcT_receives_outterm_tuple, ttg::ExecutionSpace space,
          typename keyT, typename output_terminalsT, typename... input_valuesT>
struct CallableWrapTTAsTypelist<funcT, returnT, funcT_receives_input_tuple,
                                    funcT_receives_outterm_tuple, space, keyT, output_terminalsT,
                                    std::tuple<input_valuesT...>> {
  using type = CallableWrapTT<funcT, returnT, funcT_receives_input_tuple,
                                  funcT_receives_outterm_tuple, space, keyT, output_terminalsT,
                                  std::remove_reference_t<input_valuesT>...>;
};
 
template <typename funcT, typename returnT, bool funcT_receives_input_tuple,
          bool funcT_receives_outterm_tuple, ttg::ExecutionSpace space,
          typename keyT, typename output_terminalsT, typename... input_valuesT>
struct CallableWrapTTAsTypelist<funcT, returnT, funcT_receives_input_tuple,
                                    funcT_receives_outterm_tuple, space, keyT, output_terminalsT,
                                    ttg::meta::typelist<input_valuesT...>> {
  using type = CallableWrapTT<funcT, returnT, funcT_receives_input_tuple,
                                  funcT_receives_outterm_tuple, space, keyT, output_terminalsT,
                                  std::remove_reference_t<input_valuesT>...>;
};
 
// clang-format off
// clang-format on
template <ttg::ExecutionSpace space,
          typename keyT = void,
          typename funcT,
          typename... input_edge_valuesT,
          typename... output_edgesT>
auto make_tt_tpl(funcT &&func, const std::tuple<ttg::Edge<keyT, input_edge_valuesT>...> &inedges = std::tuple<>{},
                 const std::tuple<output_edgesT...> &outedges = std::tuple<>{}, const std::string &name = "wrapper",
                 const std::vector<std::string> &innames = std::vector<std::string>(sizeof...(input_edge_valuesT),
                                                                                    "input"),
                 const std::vector<std::string> &outnames = std::vector<std::string>(sizeof...(output_edgesT),
                                                                                     "output")) {
  // ensure input types do not contain Void
  static_assert(ttg::meta::is_none_Void_v<input_edge_valuesT...>, "ttg::Void is for internal use only, do not use it");
  using output_terminals_type = typename ttg::edges_to_output_terminals<std::tuple<output_edgesT...>>::type;
 
  constexpr auto void_key = ttg::meta::is_void_v<keyT>;
 
  // list of base datum types (T or const T)
  using base_input_data_t = ttg::meta::typelist<typename ttg::Edge<keyT, input_edge_valuesT>::value_type...>;
 
  // gross list of candidate argument types
  using gross_candidate_func_args_t = ttg::meta::typelist<
      ttg::meta::candidate_argument_bindings_t<std::add_const_t<keyT>>,
      ttg::meta::candidate_argument_bindings_t<
          std::tuple<std::add_lvalue_reference_t<typename ttg::Edge<keyT, input_edge_valuesT>::value_type>...>>,
      ttg::meta::typelist<output_terminals_type &>>;
 
  // net list of candidate argument types excludes the empty typelists for void arguments
  using candidate_func_args_t = ttg::meta::filter_t<gross_candidate_func_args_t, ttg::meta::typelist_is_not_empty>;
 
  // compute list of argument types with which func can be invoked
  constexpr static auto func_is_generic = ttg::meta::is_generic_callable_v<funcT>;
  using return_type_typelist_and_gross_func_args_t =
      decltype(ttg::meta::compute_arg_binding_types(func, candidate_func_args_t{}));
  using func_return_t = std::tuple_element_t<0, std::tuple_element_t<0, return_type_typelist_and_gross_func_args_t>>;
  using gross_func_args_t = std::tuple_element_t<1, return_type_typelist_and_gross_func_args_t>;
  constexpr auto DETECTED_HOW_TO_INVOKE_GENERIC_FUNC =
      func_is_generic ? !std::is_same_v<gross_func_args_t, ttg::typelist<>> : true;
  static_assert(DETECTED_HOW_TO_INVOKE_GENERIC_FUNC,
                "ttd::make_tt_tpl(func, inedges, ...): could not detect how to invoke generic callable func, either "
                "the signature of func "
                "is faulty, or inedges does match the expected list of types, or both");
 
  // net argument typelist
  using func_args_t = ttg::meta::drop_void_t<gross_func_args_t>;
  constexpr auto num_args = std::tuple_size_v<func_args_t>;
 
  // if given task id, make sure it's passed via const lvalue ref
  constexpr bool TASK_ID_PASSED_AS_CONST_LVALUE_REF =
      !void_key ? ttg::meta::probe_first_v<ttg::meta::is_const_lvalue_reference, true, func_args_t> : true;
  constexpr bool TASK_ID_PASSED_AS_NONREF =
      !void_key ? !ttg::meta::probe_first_v<std::is_reference, true, func_args_t> : true;
  static_assert(
      TASK_ID_PASSED_AS_CONST_LVALUE_REF || TASK_ID_PASSED_AS_NONREF,
      "ttg::make_tt_tpl(func, ...): if given to func, the task id must be passed by const lvalue ref or by value");
 
  // if given out-terminal tuple, make sure it's passed via nonconst lvalue ref
  constexpr bool have_outterm_tuple =
      func_is_generic ? !ttg::meta::is_last_void_v<gross_func_args_t>
                      : ttg::meta::probe_last_v<ttg::meta::is_nonconst_lvalue_reference_to_output_terminal_tuple, true,
                                                gross_func_args_t>;
  constexpr bool OUTTERM_TUPLE_PASSED_AS_NONCONST_LVALUE_REF =
      have_outterm_tuple ? ttg::meta::probe_last_v<ttg::meta::is_nonconst_lvalue_reference, true, func_args_t> : true;
  static_assert(
      OUTTERM_TUPLE_PASSED_AS_NONCONST_LVALUE_REF,
      "ttd::make_tt_tpl(func, ...): if given to func, the output terminal tuple must be passed by nonconst lvalue ref");
 
  static_assert(num_args == 3 - (void_key ? 1 : 0) - (have_outterm_tuple ? 0 : 1),
                "ttg::make_tt_tpl(func, ...): func takes wrong number of arguments (2, or 1, if keyT=void + optional "
                "tuple of output terminals)");
 
  // 2. input_args_t = {input_valuesT&&...}
  using nondecayed_input_args_t = std::tuple_element_t<void_key ? 0 : 1, func_args_t>;
  constexpr auto NO_ARGUMENTS_PASSED_AS_NONCONST_LVALUE_REF =
      !ttg::meta::is_any_nonconst_lvalue_reference_v<nondecayed_input_args_t>;
  static_assert(
      NO_ARGUMENTS_PASSED_AS_NONCONST_LVALUE_REF,
      "ttg::make_tt_tpl(func, inedges, outedges): one or more arguments to func can only be passed by nonconst lvalue "
      "ref; this is illegal, should only pass arguments as const lvalue ref or (nonconst) rvalue ref");
  using input_args_t = std::decay_t<nondecayed_input_args_t>;
  using decayed_input_args_t = ttg::meta::decayed_typelist_t<input_args_t>;
  using wrapT = typename CallableWrapTTAsTypelist<funcT, func_return_t, true, have_outterm_tuple, space, keyT,
                                                      output_terminals_type, input_args_t>::type;
  static_assert(std::is_same_v<decayed_input_args_t, std::tuple<input_edge_valuesT...>>,
                "ttg::make_tt_tpl(func, inedges, outedges): inedges value types do not match argument types of func");
 
  return std::make_unique<wrapT>(std::forward<funcT>(func), inedges, outedges, name, innames, outnames);
}
 
template <typename keyT = void,
          typename funcT,
          typename... input_edge_valuesT,
          typename... output_edgesT>
auto make_tt_tpl(funcT &&func, const std::tuple<ttg::Edge<keyT, input_edge_valuesT>...> &inedges = std::tuple<>{},
                 const std::tuple<output_edgesT...> &outedges = std::tuple<>{}, const std::string &name = "wrapper",
                 const std::vector<std::string> &innames = std::vector<std::string>(sizeof...(input_edge_valuesT),
                                                                                    "input"),
                 const std::vector<std::string> &outnames = std::vector<std::string>(sizeof...(output_edgesT),
                                                                                     "output"))
{
  return make_tt_tpl<ttg::ExecutionSpace::Host, keyT>(
      std::forward<funcT>(func), inedges, outedges, name, innames, outnames);
}
// clang-format off
// clang-format on
template <ttg::ExecutionSpace space,
          typename keyT = void, typename funcT,
          typename... input_edge_valuesT, typename... output_edgesT>
auto make_tt(funcT &&func, const std::tuple<ttg::Edge<keyT, input_edge_valuesT>...> &inedges = std::tuple<>{},
             const std::tuple<output_edgesT...> &outedges = std::tuple<>{}, const std::string &name = "wrapper",
             const std::vector<std::string> &innames = std::vector<std::string>(sizeof...(input_edge_valuesT), "input"),
             const std::vector<std::string> &outnames = std::vector<std::string>(sizeof...(output_edgesT), "output")) {
  // ensure input types do not contain Void
  static_assert(ttg::meta::is_none_Void_v<input_edge_valuesT...>, "ttg::Void is for internal use only, do not use it");
 
  using output_terminals_type = typename ttg::edges_to_output_terminals<std::tuple<output_edgesT...>>::type;
 
  constexpr auto void_key = ttg::meta::is_void_v<keyT>;
 
  // list of base datum types (T or const T)
  using base_input_data_t = ttg::meta::typelist<typename ttg::Edge<keyT, input_edge_valuesT>::value_type...>;
 
  // gross list of candidate argument types
  using gross_candidate_func_args_t = ttg::meta::typelist<
      ttg::meta::candidate_argument_bindings_t<std::add_const_t<keyT>>,
      ttg::meta::candidate_argument_bindings_t<typename ttg::Edge<keyT, input_edge_valuesT>::value_type>...,
      ttg::meta::typelist<output_terminals_type &, void>>;
 
  // net list of candidate argument types excludes the empty typelists for void arguments
  using candidate_func_args_t = ttg::meta::filter_t<gross_candidate_func_args_t, ttg::meta::typelist_is_not_empty>;
 
  // gross argument typelist for invoking func, can include void for optional args
  constexpr static auto func_is_generic = ttg::meta::is_generic_callable_v<funcT>;
  using return_type_typelist_and_gross_func_args_t =
      decltype(ttg::meta::compute_arg_binding_types(func, candidate_func_args_t{}));
  using func_return_t = std::tuple_element_t<0, std::tuple_element_t<0, return_type_typelist_and_gross_func_args_t>>;
  using gross_func_args_t = std::tuple_element_t<1, return_type_typelist_and_gross_func_args_t>;
  constexpr auto DETECTED_HOW_TO_INVOKE_GENERIC_FUNC =
      func_is_generic ? !std::is_same_v<gross_func_args_t, ttg::typelist<>> : true;
  static_assert(DETECTED_HOW_TO_INVOKE_GENERIC_FUNC,
                "ttd::make_tt(func, inedges, ...): could not detect how to invoke generic callable func, either the "
                "signature of func "
                "is faulty, or inedges does match the expected list of types, or both");
 
  // net argument typelist
  using func_args_t = ttg::meta::drop_void_t<gross_func_args_t>;
  constexpr auto num_args = std::tuple_size_v<func_args_t>;
 
  // if given task id, make sure it's passed via const lvalue ref
  constexpr bool TASK_ID_PASSED_AS_CONST_LVALUE_REF =
      !void_key ? ttg::meta::probe_first_v<ttg::meta::is_const_lvalue_reference, true, func_args_t> : true;
  constexpr bool TASK_ID_PASSED_AS_NONREF =
      !void_key ? !ttg::meta::probe_first_v<std::is_reference, true, func_args_t> : true;
  static_assert(
      TASK_ID_PASSED_AS_CONST_LVALUE_REF || TASK_ID_PASSED_AS_NONREF,
      "ttg::make_tt(func, ...): if given to func, the task id must be passed by const lvalue ref or by value");
 
  // if given out-terminal tuple, make sure it's passed via nonconst lvalue ref
  constexpr bool have_outterm_tuple =
      func_is_generic ? !ttg::meta::is_last_void_v<gross_func_args_t>
                      : ttg::meta::probe_last_v<ttg::meta::decays_to_output_terminal_tuple, false, gross_func_args_t>;
  constexpr bool OUTTERM_TUPLE_PASSED_AS_NONCONST_LVALUE_REF =
      have_outterm_tuple ? ttg::meta::probe_last_v<ttg::meta::is_nonconst_lvalue_reference, false, func_args_t> : true;
  static_assert(
      OUTTERM_TUPLE_PASSED_AS_NONCONST_LVALUE_REF,
      "ttg::make_tt(func, ...): if given to func, the output terminal tuple must be passed by nonconst lvalue ref");
 
  // TT needs actual types of arguments to func ... extract them and pass to CallableWrapTT
  using input_edge_value_types = ttg::meta::typelist<std::decay_t<input_edge_valuesT>...>;
  // input_args_t = {input_valuesT&&...}
  using input_args_t = typename ttg::meta::take_first_n<
      typename ttg::meta::drop_first_n<func_args_t, std::size_t(void_key ? 0 : 1)>::type,
      std::tuple_size_v<func_args_t> - (void_key ? 0 : 1) - (have_outterm_tuple ? 1 : 0)>::type;
  constexpr auto NO_ARGUMENTS_PASSED_AS_NONCONST_LVALUE_REF =
      !ttg::meta::is_any_nonconst_lvalue_reference_v<input_args_t>;
  static_assert(
      NO_ARGUMENTS_PASSED_AS_NONCONST_LVALUE_REF,
      "ttg::make_tt(func, inedges, outedges): one or more arguments to func can only be passed by nonconst lvalue "
      "ref; this is illegal, should only pass arguments as const lvalue ref or (nonconst) rvalue ref");
  using decayed_input_args_t = ttg::meta::decayed_typelist_t<input_args_t>;
  // 3. full_input_args_t = edge-types with non-void types replaced by input_args_t
  using full_input_args_t = ttg::meta::replace_nonvoid_t<input_edge_value_types, input_args_t>;
  using wrapT = typename CallableWrapTTAsTypelist<funcT, func_return_t, false, have_outterm_tuple, space, keyT,
                                                      output_terminals_type, full_input_args_t>::type;
 
  return std::make_unique<wrapT>(std::forward<funcT>(func), inedges, outedges, name, innames, outnames);
}
 
template <typename keyT = void, typename funcT,
          typename... input_edge_valuesT, typename... output_edgesT>
auto make_tt(funcT &&func, const std::tuple<ttg::Edge<keyT, input_edge_valuesT>...> &inedges = std::tuple<>{},
             const std::tuple<output_edgesT...> &outedges = std::tuple<>{}, const std::string &name = "wrapper",
             const std::vector<std::string> &innames = std::vector<std::string>(sizeof...(input_edge_valuesT), "input"),
             const std::vector<std::string> &outnames = std::vector<std::string>(sizeof...(output_edgesT), "output")) {
  return make_tt<ttg::ExecutionSpace::Host, keyT>(std::forward<funcT>(func), inedges, outedges, name, innames, outnames);
}
 
template <typename keyT, typename funcT, typename... input_valuesT, typename... output_edgesT>
[[deprecated("use make_tt_tpl instead")]] inline auto wrapt(
    funcT &&func, const std::tuple<ttg::Edge<keyT, input_valuesT>...> &inedges,
    const std::tuple<output_edgesT...> &outedges, const std::string &name = "wrapper",
    const std::vector<std::string> &innames = std::vector<std::string>(sizeof...(input_valuesT), "input"),
    const std::vector<std::string> &outnames = std::vector<std::string>(sizeof...(output_edgesT), "output")) {
  return make_tt_tpl<keyT>(std::forward<funcT>(func), inedges, outedges, name, innames, outnames);
}
 
template <typename keyT, typename funcT, typename... input_edge_valuesT, typename... output_edgesT>
[[deprecated("use make_tt instead")]] auto wrap(
    funcT &&func, const std::tuple<ttg::Edge<keyT, input_edge_valuesT>...> &inedges,
    const std::tuple<output_edgesT...> &outedges, const std::string &name = "wrapper",
    const std::vector<std::string> &innames = std::vector<std::string>(sizeof...(input_edge_valuesT), "input"),
    const std::vector<std::string> &outnames = std::vector<std::string>(sizeof...(output_edgesT), "output")) {
  return make_tt<keyT>(std::forward<funcT>(func), inedges, outedges, name, innames, outnames);
}
 
#endif  // TTG_MAKE_TT_H