c++ - 双模板函数实例化失败

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
}