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
}

触发以下错误信息:

error: an explicit template argument list is not allowed on this declaration

为什么？

注意事项:

• 我很确定这与 CUDA 无关，只是一个 C++ 问题。
• 有很多关于偏特化的问题，但我不知道我的问题是否是其中任何一个的骗局。

Answer 1

您不能对模板函数进行部分特化，因为 C++ 没有定义这样的东西。您只需执行 class 模板部分特化 [§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
}