这是一个最小的例子
#include <iostream>
template<typename T>
struct MyClass{
T value;
};
template<typename T, template<typename> class Class>
void foo(Class<T>& myClass){
myClass.value=0;
};
int main(){
MyClass<double> myclass;
foo<double, MyClass<double>>(myclass);
return 0;
}
此代码无法编译并提供错误
error: no instance of function template "foo" matches the argument list
argument types are: (MyClass<double>)
foo<double, MyClass<double>>(myclass);
我想编写函数的原因是我想编写一个在CPU和GPU之间传输数据的函数。该函数看起来像
template<typename Scalar, template<typename > class Host,
template<typename> class Device>
void copyFromHostAsync(const Host<Scalar> &host, Device<Scalar> &device, cudaStream_t stream) {
assert(host.rows() == device.rows());
assert(host.cols() == device.cols());
cudaMemcpyAsync(device.getPtr(), host.getPtr(), sizeof(Scalar) * host.rows() * host.cols(),
cudaMemcpyHostToDevice, stream);
}
我想使用模板化的类作为参数,以便底层标量类型相同。
欢迎任何帮助。
答案 0 :(得分:2)
foo
是一个模板函数,它将模板参数作为类型T
和一个参数类型为MyClass
的模板类型。
如果你写:
foo<double, MyClass<double>>(myclass);
第二个模板参数是不具有1个参数类型的模板。这只是一种简单的类型。因为它是一种类型,而不是模板类型,所以您的代码无法编译。
仅使用MyClass
将编译,因为MyClass
是一个模板类型,它带有1个模板参数:
foo<double, MyClass>(myclass);
另外,让编译器为您完成工作并让它推断出类型:
foo(myclass);