0

我从 OpenACC 计算区域调用了一个 cuda 函数，我想指定应该进入 cuda 函数的线程数，但似乎我不知道如何控制它。

%main.cpp
..

#pragma acc routine vector
extern "C" void CUDA_KERNEL_FUNCTION(double *B, int ldb,const double *A, int lda);
..
#pragma acc parallel loop independent collapse(3) gang vector(128)
  for(int i0 = 0; i0 < size0 - 31; i0+= 32)
     for(int i1 = 0; i1 < size1 - 31; i1+= 32)
        for(int i2 = 0; i2 < size2; i2+= 1)
          CUDA_KERNEL_FUNCTION(B, ldb, A, lda);

..
..


%cuda_code.cu

extern "C" __device__ void CUDA_KERNEL_FUNCTION(double *B, int ldb,const double *A, int lda)
{

    Num_Threads_gpu = blockDim.x * blockDim.y* blockDim.z;

    //Num_Threads_gpu is always 32 
}

编译很好。但是无论我使用什么向量长度，进入 cuda 函数的线程数始终是 32。有没有办法指定它？

我使用“cuda/7.0.28”和“pgi/15.10”

谢谢

4

1 回答 1

1

尝试将 vector(128) 更改为 vector_length(128)。我认为 PGI 15.10 支持这两种语法，但以防万一……

如果这不起作用，您能否发布编译器输出，-Minfo=accel以便我们可以看到编译器在做什么？

于 2015-11-10T22:28:59.027 回答