专门针对 C++AMP 受限 lambda 的模板

Question

使用this question（和其他一些人）的洞察力，我已经能够编写以下代码来询问正常的 lambda 函数类型信息（即返回类型、参数计数等）

// helper classes ========================================
template <typename R, typename... A>
class lambda_traits_evaluation {
public:
  typedef R r_type;
  enum { n_args = sizeof...(A) };
  // ...
};

template <typename R, typename... A>
class lambda_traits_helper
  : public lambda_traits_evaluation<R,A...>{};

template <typename R, typename F, typename... A>
class lambda_traits_helper<R (F::*)(A...) const>
  : public lambda_traits_evaluation<R,A...>{};

// use class      ========================================
template <typename F>
class lambda_traits {
  typedef typename lambda_traits_helper<decltype(&F::operator())> helper_impl;
  // ...
}

然后我可以将它与lambda_traits<decltype(myLambda)> 一起使用，但这就是我自鸣得意的编码结束的地方，因为如果我的 lambda 是 gpu 的放大器限制，即

auto myLambda = [](int) restrict(amp) -> void {};

显然模板专业化没有被选中。但是添加新的专业化

template <typename R, typename F, typename... A>
class lambda_traits_helper<R (F::*)(A...) const restrict(amp)>
  : public lambda_traits_evaluation<R,A...> {};

仍然没有解决问题，因为我发现编译器吠叫

error C3939: 'abstract declarator' : pointer to member functions, function
             pointers, references to functions with 'amp' restriction
             specifier are not allowed

是否有另一种方法来询问 lambdas 中的类型，或者有一种方法可以去除 lambda 类型的限制？

Answer 1

即使在未评估的上下文中，也无法形成指向受 amp 限制的函数的指针，这真是令人遗憾。但是，有一种解决方法，只要您可以要求 amp-restricted lambda 为 cpu,amp-restricted，这是可行的。在这种情况下，您可以抛弃 amp-restriction，形成一个指向 cpu-restricted 成员函数的指针——您可以进一步询问。

请参阅以下概念验证：

#include <type_traits>

template <typename R, typename F, typename... A>
auto get_R(R (F::*)(A...) const) -> R
{}

template <typename L>
struct lambda_traits
{
    using ret_type = decltype(get_R(&L::operator()));
};

int main()
{
    auto lambda_1 = [](int) restrict(cpu,amp) -> void {};
    auto lambda_2 = [](int) restrict(cpu,amp) -> int { return 0; };

    // Test:
    static_assert(std::is_same<lambda_traits<decltype(lambda_1)>::ret_type, void>::value, "Failed 1.");
    static_assert(std::is_same<lambda_traits<decltype(lambda_2)>::ret_type, int>::value, "Failed 2.");
}

希望有帮助！