OpenACC - 私人二维阵列

debugcn 发表于 Dev

随机游客

我正在处理块对角矩阵（每个块具有相同的大小），illegal address error当我使用private动态分配的二维数组时，我有一个...

// NB is the number of block
// N is the block size
// A is the main matrix (block diagonal)

double** B; // a block
B = new double*[N];
for (unsigned int i = 0; i < N; i++)
    B[i] = new double[N];

#pragma acc parallel loop private(B[:N][:N]) copyin(A[:NB*N][:NB*N])
for (unsigned int b = 0; b < NB; b++) {
    #pragma acc loop
    for (unsigned int i = 0; i < N; i++) {
        #pragma acc loop
        for (unsigned int j = 0; j < N; j++) {
            B[i][j] = A[b*N+i][b*N+j];
        }
    }
    // process B
}

for (unsigned int i = 0; i < N; i++)
    delete[] B[i];
delete[] B;

我得到的错误是：

call to cuStreamSynchronize returned error 700: Illegal address during kernel execution

如果我将数组展平为一维数组并使用字典索引或静态二维数组，它工作正常，但我使用的函数需要double**作为参数，所以我更愿意坚持使用动态二维数组...

我已经阅读了private规范中的条款，但它并没有说不支持动态二维数组，所以我想我做错了什么......

马特·科尔格罗夫

抱歉，不支持在私有子句中使用指针数组。问题是编译器运行时必须为每个帮派、工人或向量动态创建一个私有的（取决于带有私有子句的循环）并填充所有设备指针。这将带来极高的间接成本。

如果“B”是一个固定大小的静态数组，“double B[N][N]”，那么你可以在 private 子句中使用它。

否则，我建议通过添加第三维来手动私有化数组。

 // NB is the number of block
// N is the block size
// A is the main matrix (block diagonal)

double*** B; // a block
B = new double**[NB];
for (unsigned int i = 0; i < NB; i++) {
  B[i] = new double*[N];
  for (unsigned int j = 0; j < N; j++) {
    B[i][j] = new double[N];
}}


#pragma acc parallel loop create(B[:NB][:N][:N]) copyin(A[:NB*N][:NB*N])
for (unsigned int b = 0; b < NB; b++) {
    #pragma acc loop
    for (unsigned int i = 0; i < N; i++) {
        #pragma acc loop
        for (unsigned int j = 0; j < N; j++) {
            B[b][i][j] = A[b*N+i][b*N+j];
        }
    }
    // process B
}

for (unsigned int i = 0; i < NB; i++) {
  for (unsigned int j = 0; j < N; j++) {
     delete[] B[i][j];
  }
  delete[] B[i];
}
delete[] B;

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。