我必须编写一个获取两个矩阵 Nx3 和 Mx3 的 PyCUDA 函数,并返回一个矩阵 NxM,但我不知道如何在不知道列数的情况下通过引用传递矩阵。
我的代码基本上是这样的:
#kernel declaration
mod = SourceModule("""
__global__ void distance(int N, int M, float d1[][3], float d2[][3], float res[][M])
{
int i = threadIdx.x;
int j = threadIdx.y;
float x, y, z;
x = d2[j][0]-d1[i][0];
y = d2[j][1]-d1[i][1];
z = d2[j][2]-d1[i][2];
res[i][j] = x*x + y*y + z*z;
}
""")
#load data
data1 = numpy.loadtxt("data1.txt").astype(numpy.float32) # Nx3 matrix
data2 = numpy.loadtxt("data2.txt").astype(numpy.float32) # Mx3 matrix
N=data1.shape[0]
M=data2.shape[0]
res = numpy.zeros([N,M]).astype(numpy.float32) # NxM matrix
#invoke kernel
dist_gpu = mod.get_function("distance")
dist_gpu(cuda.In(numpy.int32(N)), cuda.In(numpy.int32(M)), cuda.In(data1), cuda.In(data2), cuda.Out(res), block=(N,M,1))
#save data
numpy.savetxt("results.txt", res)
编译这个我收到一个错误:
kernel.cu(3): error: a parameter is not allowed
也就是说,我不能在函数声明中使用 M 作为 res[][] 的列数。我不能留下未声明的列数......
我需要一个矩阵 NxM 作为输出,但我不知道该怎么做。你能帮助我吗?