mmdeploy/csrc/execution/utility.h
lzhangzz 46bfe0ac87
[Feature] New pipeline & executor for SDK (#497)
* executor prototype

* add split/when_all

* fix GCC build

* WIP let_value

* fix let_value

* WIP ensure_started

* ensure_started & start_detached

* fix let_value + when_all combo on MSVC 142

* fix static thread pool

* generic just, then, let_value, sync_wait

* minor

* generic split and when_all

* fully generic sender adapters

* when_all: workaround for GCC7

* support legacy spdlog

* fix memleak

* bulk

* static detector

* fix bulk & first pipeline

* bulk for static thread pools

* fix on MSVC

* WIP async batch submission

* WIP collation

* async batch

* fix detector

* fix async detector

* fix

* fix

* debug

* fix cuda allocator

* WIP type erased executor

* better type erasure

* simplify C API impl

* Expand & type erase TC

* deduction guide for type erased senders

* fix GCC build

* when_all for arrays of Value senders

* WIP pipeline v2

* WIP pipeline parser

* WIP timed batch operation

* add registry

* experiment

* fix pipeline

* naming

* fix mem-leak

* fix deferred batch operation

* WIP

* WIP configurable scheduler

* WIP configurable scheduler

* add comment

* parse scheduler config

* force link schedulers

* WIP pipeable sender

* WIP CPO

* ADL isolation and dismantle headers

* type erase single thread context

* fix MSVC build

* CPO

* replace decay_t with remove_cvref_t

* structure adjustment

* structure adjustment

* apply CPOs & C API rework

* refine C API

* detector async C API

* adjust detector async C API

* # Conflicts:
#	csrc/apis/c/detector.cpp

* fix when_all for type erased senders

* support void return for Then

* async detector

* fix some CPOs

* minor

* WIP rework capture mechanism for type erased types

* minor fix

* fix MSVC build

* move expand.h to execution

* make `Expand` pipeable

* fix type erased

* un-templatize `_TypeErasedOperation`

* re-work C API

* remove async_detector C API

* fix pipeline

* add flatten & unflatten

* fix flatten & unflatten

* add aync OCR demo

* config executor for nodes & better executor API

* working async OCR example

* minor

* dynamic batch via scheduler

* dynamic batch on `Value`

* fix MSVC build

* type erase dynamic batch scheduler

* sender as Python Awaitable

* naming

* naming

* add docs

* minor

* merge tmp branch

* unify C APIs

* fix ocr

* unify APIs

* fix typo

* update async OCR demo

* add v3 API text recognizer

* fix v3 API

* fix lint

* add license info & reformat

* add demo async_ocr_v2

* revert files

* revert files

* resolve link issues

* fix scheduler linkage for shared libs

* fix license header

* add docs for `mmdeploy_executor_split`

* add missing `mmdeploy_executor_transfer_just` and `mmdeploy_executor_execute`

* make `TimedSingleThreadContext` header only

* fix lint

* simplify type-erased sender
2022-06-01 14:10:43 +08:00

85 lines
2.3 KiB
C++

// Copyright (c) OpenMMLab. All rights reserved.
// Modified from
// https://github.com/brycelelbach/wg21_p2300_std_execution/blob/main/include/__utility.hpp
#ifndef MMDEPLOY_CSRC_EXPERIMENTAL_EXECUTION_UTILITY_H_
#define MMDEPLOY_CSRC_EXPERIMENTAL_EXECUTION_UTILITY_H_
#include <type_traits>
#include <utility>
#include "core/mpl/detected.h"
#include "tag_invoke.h"
#define MMDEPLOY_REQUIRES(...) std::enable_if_t<__VA_ARGS__, int> = 0
namespace mmdeploy {
template <typename T, typename E, typename U = void>
using _decays_to = std::enable_if_t<std::is_same<std::decay_t<T>, E>::value, U>;
template <typename... Ts>
using __decayed_tuple = std::tuple<std::decay_t<Ts>...>;
template <typename Fun, typename... As>
using __call_result_t = decltype(std::declval<Fun>()(std::declval<As>()...));
template <typename F>
struct __conv {
F f_;
using type = __call_result_t<F>;
operator type() && { return ((F &&) f_)(); }
};
template <typename F>
__conv(F)->__conv<F>;
template <typename T, typename = std::enable_if_t<std::is_destructible_v<T>>>
struct __conv_proxy {
T v_;
template <typename F>
explicit __conv_proxy(F&& f) : v_(((F &&) f)()) {}
T& operator*() noexcept { return v_; }
};
template <typename _Member, typename _Self>
_Member _Self::*__memptr(const _Self&);
template <typename _Self, typename _Member>
using __member_t = decltype((std::declval<_Self>().*__memptr<_Member>(std::declval<_Self>())));
template <typename From, typename To>
using _copy_cvref_t = __member_t<From, To>;
template <typename S, typename R>
using connect_result_t = decltype(Connect(std::declval<S>(), std::declval<R>()));
template <typename...>
struct _types
#if defined(__GNUC__) && !defined(__clang__)
{
}
#endif
;
namespace __schedule {
struct schedule_t {
template <typename Scheduler, std::enable_if_t<tag_invocable<schedule_t, Scheduler>, int> = 0>
auto operator()(Scheduler&& scheduler) const -> tag_invoke_result_t<schedule_t, Scheduler> {
return tag_invoke(schedule_t{}, (Scheduler &&) scheduler);
}
};
} // namespace __schedule
using __schedule::schedule_t;
inline constexpr schedule_t Schedule{};
template <typename Scheduler>
using schedule_result_t = decltype(std::declval<schedule_t>()(std::declval<Scheduler>()));
} // namespace mmdeploy
#endif // MMDEPLOY_CSRC_EXPERIMENTAL_EXECUTION_UTILITY_H_