Question

以下代码：

template<typename T, MyEnum K> __global__ void myKernel(const T a[]);
template<typename T> __global__ void myKernel<T,SomeValueOfMyEnum>(const T a[]) {
    // implementation
}

触发以下错误消息：

错误：此声明
上不允许使用显式模板参数列表

为什么呢？

注意：

我很确定这不是CUDA相关的，只是一个C ++问题。
关于部分专业化有很多问题，但我无法弄清楚我的是否是他们中的任何一个。

Answer 1

你不能对模板函数进行部分特化，因为C ++没有定义这样的东西。你可以做一个类模板部分特化[§14.5.5/ temp.class.spec]

类部分专业化 - 有点难看，但也许它可以帮助你。

enum MyEnum
{
    E1, E2
};

template<typename T, MyEnum K>
struct MyKernel
{
    void operator()(const T a[])
    {
        // ...
    }
};

template<typename T>
struct MyKernel<T, E1>
{
    void operator()(const T a[])
    {
        // ...
    }
};

int main()
{
    MyKernel<int, E1>()( ... ); // <--- To call
}

Answer 2

您可以使用enable_if实现目标。

//template<typename T, MyEnum K> __global__ void myKernel(const T a[]);

template<typename T, MyEnum K>
typename std::enable_if<std::is_same<K, SomeValueOfMyEnum>::value, __global__ void>::type
myKernel<T,SomeValueOfMyEnum>(const T a[]) 
{
     // implementation
}

template<typename T, MyEnum K>
typename std::enable_if<!std::is_same<K, SomeValueOfMyEnum>::value, __global__ void>::type
myKernel<T,SomeValueOfMyEnum>(const T a[]) 
{
     // implementation
}

双模板函数实例化失败

2 个答案: