CUDA内核"只允许单个包参数"解决方法吗?

时间:2017-05-19 18:18:54

标签: c++ c++11 templates cuda variadic-templates

关于可变参数全局函数模板的CUDA 7标准说明"only a single pack parameter is allowed."这是否有一个优雅的解决方法?我希望能够做到这样的事情:

template<int... vals>
void RecursiveFunct() {

}

template<int... vals, typename T, typename... Args>
void RecursiveFunct(T t, Args... args) {
  t.template call<vals...>();
  RecursiveFunct<vals...>(args...);
}

我认为我可以在传递它们之前将我的整数包装成一些东西但是是否可以以对此代码的调用者透明的方式执行此操作?

1 个答案:

答案 0 :(得分:3)

不确定理解您的确切限制但是...我认为std::integer_sequence和调用call()的包装函数可以帮助您。

以下是一个玩具,但可编辑,显示我的意思。

struct foo
 {
   template <int ... vals>
   void call () const
    { std::cout << "- call " << sizeof...(vals) << std::endl; }
 };

template <typename IS>
void RecursiveFunct (IS const &)
 { }

template <typename T, int ... vals>
void wrapCall (T const & t, std::integer_sequence<int, vals...> const &)
 { t.template call<vals...>(); }


template<typename IS,  typename T, typename ... Args>
void RecursiveFunct (IS const & is, T t, Args... args)
 {
   wrapCall(t, is);
   RecursiveFunct(is, args...);
 }

int main ()
 {
   // print 5 times "- call 4"
   RecursiveFunct(std::integer_sequence<int, 2, 3, 5, 7>{},
                  foo{}, foo{}, foo{}, foo{}, foo{});
 }

请记住std::integer_sequence是C ++ 14的特性,因此前面的代码需要(至少)一个C ++ 14编译器。

但是如果你需要使用C ++ 11,那么创建一个std::integer_sequence替代品是微不足道的。

以示例

template <typename T, T ... ts>
struct myIntegerSequence
 { };

- 编辑 -

OP问

  

这可以在不创建integer_sequence实例的情况下工作吗?

在普通的C ++ 14中,是的。和Cuda合作吗?我不知道。

我已经通过wrapCall()结构和wrapCall静态方法更改了func() func。这是因为我使用了不能与funcs一起使用的部分特化。

愚蠢是玩具的例子

#include <utility>
#include <iostream>

struct foo
 {
   template <int ... vals>
   void call () const
    { std::cout << "- call " << sizeof...(vals) << std::endl; }
 };

template <typename>
void RecursiveFunct ()
 { }

template <typename>
struct wrapCall;

template <int ... vals>
struct wrapCall<std::integer_sequence<int, vals...>>
 {
   template <typename T>
   static constexpr void func (T const & t)
    { t.template call<vals...>(); }
 };

template<typename IS,  typename T, typename ... Args>
void RecursiveFunct (T t, Args... args)
 {
   wrapCall<IS>::func(t);
   RecursiveFunct<IS>(args...);
 }

int main ()
 {
   // print 5 times "- call 4"
   RecursiveFunct<std::integer_sequence<int, 2, 3, 5, 7>>
                  (foo{}, foo{}, foo{}, foo{}, foo{});
 }

但是你确定std::integer_sequence

的问题是否存在问题?