在C++中,不可能通过
namespace
因为某种类型的参数(通过模板或实际函数参数)对类或函数是不可能的。这同样适用于CUDA(至少据我所知)。在这个问题中解释了一些原因:
Why can't namespaces be template parameters?
下面是一个用例示例:
namespace experiment1
{
int repetitions() { return 2; }
void setup() { ... }
void f() { ... }
void teardown() { ... }
}
namespace experiment2
{
int repetitions() { return 4; }
void setup() { ... }
void f() { ... }
void teardown() { ... }
}
// Beware, this is invalid C++ and invalid CUDA
template<namespace NS>
void do_test()
{
// Do something with NS::repetitions(), NS::setup(), ...
}
这在C++中无效的原因之一是,在这种方法中,没有什么是不能用类实现的。您确实可以将每个命名空间转换为类,将函数转换为成员函数,然后将类作为模板参数传递给
do_test
函数或其实例作为同一函数的参数(在前一种情况下可能使用静态函数,在后一种情况中可能使用虚拟函数)。
我同意这一点。然而,在CUDA的特定情况下,可以使用名称空间,但不能使用类。想象
f
是内核,即
__global__
功能,并且
setup
或者使用另一个函数来指定例如要分配给内核的共享内存的大小。内核不能是类的成员(请参见此问题的答案:
Can CUDA kernels be virtual functions?
). 但是,您可以将其与与同一实验相关的其他函数一起放入
命名空间
.
考虑上面代码中显示的情况:
do_测试
是设置计时器、准备一些输入、检查输出、测量时间和执行其他操作的功能。每个实验都是一组具有相同名称和相同接口的几个函数,其中一个是内核。你想要的
do_测试
足够通用以处理所有这些实验。您希望每个实验的代码都是以某种封装形式包含的,如命名空间、结构、类等,。。。
这个问题能解决吗?
根据塔伦米的要求(顺便说一句,非常感谢您的评论),我将使问题更加具体。
我有几个执行类似操作的非常简单的内核。它们从一个大数组中加载值,对其应用模板操作,并将结果写入输出数组(与输入数组不同)。所谓模板操作,我指的是线程执行的操作
idx
关于输入值
idx公司
,及其相邻值(例如
idx-3
到
idx+3
). 其中最简单的内核只执行从输入到输出的复制:每个线程读取
input[idx]
并写入
output[idx]
。另一个示例是执行
output[idx] = input[idx+1] - input[idx-1]
。(我会留下一些细节,但你明白了。)
我想对这些内核进行基准测试,以便导出性能模型。对于每个内核,我还需要一个能够检查结果的主机函数。在每种情况下,我还有另一个内核,它通过优化以稍微不同的方式执行相同的操作,但从结果的角度来看是等效的。最后,我有一个主机函数,它打印内核的名称。以下是代码摘要:
namespace copy
{
std::string name() { return "copy"; }
__global__ void kernel(const float* input, float* output, int size);
__global__ void kernelOptimized(const float* input, float* output, int size);
bool check(const float* input, const float* output);
}
namespace difference
{
std::string name() { return "difference"; }
__global__ void kernel(const float* input, float* output, int size);
__global__ void kernelOptimized(const float* input, float* output, int size);
bool check(const float* input, const float* output);
}
我有个功能
do_测试
,我将其参数化为泛型:
typedef bool NameFunction(const float* input, const float* output);
typedef bool CheckFunction(const float* input, const float* output);
typedef void KernelFunction(const float* input, float* output, int size);
void do_test(NameFunction name, KernelFunction kernel1, KernelFunction kernel2, CheckFunction check)
{
// Set up input and output array
// Set up CUDA events
// Warm up kernels
// Run kernels
// Check results
// Measure time
// Do standard output
}
int main()
{
do_test<copy::name, copy::kernel, copy::kernelOptimized, copy::check>()
do_test<difference::name, difference::kernel, difference::kernelOptimized, difference::check>()
}
当然,这种方式已经很好了。然而,如果我再引入一个每个实验都必须提供的函数,我将需要修改我调用的所有这些行
do_测试
。我更希望传递此命名空间或包含这些函数的某种对象。
