0

我的问题是，当我进行乘法运算时，它仅将矩阵的第一行与向量的第一个元素相乘，而下一个元素使它们为零，因此结果向量给出了错误的结果。

using namespace std;
#define N 100
#define F 3
#define X 7
__global__ void matvec(int *MAT, int *VEC, int *SOL) {
  int bx = blockIdx.x;
  int tx = threadIdx.x;
  int i = 32*bx+tx;
  for (int j = 0; j < X; j++) {
    SOL[i] = ((MAT[i * X + j] * VEC[j]) +SOL[i]) % 2;
  }
}
int main () {
int i, j;
int MAT[N][N], VEC[N], SOL[N];
int *MAT_dev, *VEC_dev, *SOL_dev;
size_t nBytes = X * X * sizeof(int);

cout << "\t- - - - - MATRIX - - - - -\n\n";
for (i = 0; i < X; i++) {
  for (j = 0; j < X; j++) {
      cout << "Element [" << i << "][" << j << "]: ";
      cin >> MAT[i][j];
  }
 }
cout << endl << endl;
for (i = 0; i < X; i++) {
  for (j = 0; j < X; j++) {
    cout << MAT[i][j] << " ";
    if (j == (X - 1))
        cout << endl;
  }
 }
cout << endl << endl;
cout << "\t- - - - - VECTOR - - - - -\n\n";
for (i = 0; i < X; i++) {
  cout << "Element [" << i << "]: ";
  cin >> VEC[i];
}
cout << endl << endl;
for (i = 0; i < X; i++) {
  cout << VEC[i] << " ";
}
cout << endl << endl;

cudaMalloc((void**)&MAT_dev, nBytes);
cudaMalloc((void**)&VEC_dev, nBytes);
cudaMalloc((void**)&SOL_dev, nBytes);

cudaMemcpy(MAT_dev, MAT, nBytes, cudaMemcpyHostToDevice);
cudaMemcpy(VEC_dev, VEC, nBytes, cudaMemcpyHostToDevice);

dim3 dimBlock(X,X);
dim3 dimGrid(1,1);

matvec<<<dimGrid,dimBlock>>>(MAT_dev, VEC_dev, SOL_dev);

cudaMemcpy(SOL, SOL_dev, nBytes, cudaMemcpyDeviceToHost);

cout << "\t- - - - - RESULT - - - - -\n\n";
for (i = 0; i < X; i++)
{
  cout << SOL[i] << " ";
}
cout << endl << endl;

cudaFree(MAT_dev);
cudaFree(VEC_dev);
cudaFree(SOL_dev);

system("PAUSE");
return 0;
}

谢谢您的帮助

4

1 回答 1

1

这是因为 的大小MAT远大于应有的大小。基本上你需要N == X，这不应该是一个问题，因为两者在编译时都是已知的。2D 数组的内存布置在一个连续的块中，C 的主要行 - 所以在你的情况下，第一行对应于前 400 ( sizeof(int)*N) 个字节，第二行对应于第二个 400，等等。 row 被称为'stride' 但是，cudaMemcpy不知道stride 是什么或MAT已经填充了哪些元素，它只是将第一个nBytes字节从MATinto复制MAT_DEV。因为nBytesis sizeof(int)*X*Xand X== 7 <<N你的矩阵的第二行和后续行永远不会被复制。只有前 196 个字节MAT被复制，解释为什么你的第二行都是零。

于 2012-09-05T03:30:35.187 回答