双模板函数实例化失败

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
}