cxx14templatetemplate.cpp

#include <type_traits>
#include <utility>
#include <iostream>
#include <tuple>
#include <array>

using kDHCandidates = std::index_sequence<1, 2, 3>;
using kKWCandidates = std::index_sequence<1, 3, 5, 7>;

// 把REPLACE_ME删掉换成你的deep_conv即可

template <size_t A, size_t B, size_t C = 1, typename ... Args>
void REPLACE_ME(Args&&...) {  std::cout << A << ' ' << B << std::endl; }

template <std::size_t L, size_t R, typename CUTuple, size_t...Is, typename ATuple, size_t...Js>
void DoublePackInvoke(const CUTuple& cuargs, std::index_sequence<Is...>, const ATuple& args, std::index_sequence<Js...>) {
    // !!! 看这里！把线面改成
    // YOUR_CUDA_FUNC<L, R, outNHWC><<<std::get<Is>(cuargs)...>>>(std::get<Js>(args)...);
    // 调用看main函数
    REPLACE_ME<L, R>(std::get<Is>(cuargs)..., std::get<Js>(args)...);
}

template<std::size_t I=0, size_t J=0, size_t... Line, size_t... Column, typename ... Args, typename ... CUDArgs>
inline std::enable_if_t<I == sizeof...(Line), void>
ForEach1D(
        const std::index_sequence<Line...>&, const std::index_sequence<Column...>&, size_t l, size_t r, const std::tuple<Args...>&, const std::tuple<CUDArgs...>&)
{ }

template<std::size_t I=0, size_t J=0, size_t... Line, size_t... Column, typename ... Args, typename ... CUDArgs>
inline std::enable_if_t<I < sizeof...(Line), void>
ForEach1D(
        const std::index_sequence<Line...>& line, const std::index_sequence<Column...>& column, size_t l, size_t r, const std::tuple<Args...>& args, const std::tuple<CUDArgs...>& cudargs)
{
    constexpr size_t L = std::get<I>(std::make_tuple(Line...));
    constexpr size_t R = std::get<J>(std::make_tuple(Column...));
    if (l == L && r == R)
        DoublePackInvoke<L, R>(
                cudargs,
                std::make_index_sequence<sizeof...(CUDArgs)>{},
                args,
                std::make_index_sequence<sizeof...(Args)>{});
    ForEach1D<I+1, J>(line, column, l, r, args, cudargs);
}

template<std::size_t I=0, size_t J=0, size_t... Line, size_t... Column, typename ... Args, typename ... CUDArgs>
inline std::enable_if_t<J == sizeof...(Column), void>
ForEach2D(
        const std::index_sequence<Line...>&, const std::index_sequence<Column...>&, size_t l, size_t r, const std::tuple<Args...>& args, const std::tuple<CUDArgs...>& cudargs)
{ }

template<std::size_t I=0, size_t J=0, size_t... Line, size_t... Column, typename ... Args, typename ... CUDArgs>
inline std::enable_if_t<J < sizeof...(Column), void>
ForEach2D(
        const std::index_sequence<Line...>& line, const std::index_sequence<Column...>& column, size_t l, size_t r, const std::tuple<Args...>& args, const std::tuple<CUDArgs...>& cudargs)
{
    ForEach1D<I, J>(line, column, l, r, args, cudargs);
    ForEach2D<I, J+1>(line, column, l, r, args, cudargs);
}

template <size_t ... Ls, size_t ... Rs, typename ... Args, typename ... CUDArgs>
void StaticDispatcher(
        std::index_sequence<Ls...>, std::index_sequence<Rs...>, size_t l, size_t r, const std::tuple<Args...>& args, const std::tuple<CUDArgs...>& cudargs) {
    ForEach2D(kDHCandidates{}, kKWCandidates{}, l, r, args, cudargs);
}


int main() {
    size_t i, j;
    std::cin >> i >> j;
    StaticDispatcher(kDHCandidates{}, kKWCandidates{}, i, j, std::make_tuple(1, 2, 3), std::make_tuple(1, 2, 3));
}