OpenCL中

Question

创建线程的网格我写了针对OpenCL内核，我初始化一个三维阵列中的所有元素 - >我* I * I + J *∫*∫。我现在在创建一个线程网格来执行元素初始化（并发）时遇到了问题。我知道我现在使用的代码只使用3个线程，我该如何扩展？

请帮忙。我是OpenCL的新手，所以任何建议或解释可能都很方便。谢谢！

这是代码：

_kernel void initialize (
int X; 
int Y; 
int Z; 
_global float*A) { 

// Get global position in X direction 
int dirX = get_global_id(0); 
// Get global position in Y direction 
int dirY = get_global_id(1); 
// Get global position in Z direction 
int dirZ = get_global_id(2); 

int A[2000][100][4]; 
int i,j,k; 
for (i=0;i<2000;i++) 
{ 
    for (j=0;j<100;j++) 
    { 
     for (k=0;k<4;k++) 
     { 
      A[dirX*X+i][dirY*Y+j][dirZ*Z+k] = i*i*i + j*j*j; 
     } 
    } 
} 
} 

Answer 1

• 创建缓冲区来存储你的输出 'A' 在调用（主机）的代码。它作为一个指针传递给你的内核，这在上面的函数定义中是正确的。但是，您不需要在内核函数中再次声明它，因此请删除行int A[2000][100][4];。

• 可以大大简化代码。使用3D全局ID来指示3D指数在数组中的每一个工作项，你可以改变环路如下（假设对于给定的i和j，向Z的所有元素应该具有相同的值）：

  __kernel void initialize (__global float* A) { 
      // cast required so that kernel compiler knows the array dimensions 
      __global float (*a)[2000][100][4] = A; 
  
      // Get global position in X direction 
      int i = get_global_id(0); 
      // Get global position in Y direction 
      int j = get_global_id(1); 
      // Get global position in Z direction 
      int k = get_global_id(2); 
  
      (*a)[i][j][k] = i*i*i + j*j*j; 
  } 
  

在您的调用代码中，您将创建一个全局工作大小为2000x100x4的内核。

• 实际上，这是一个很多的工作项目安排，所以你可能会从全球（一维）2000年工作的大小和内核内部循环，例如，获得更好的性能：

  __kernel void initialize (__global float* A) { 
      // cast required so that kernel compiler knows the array dimensions 
      __global float (*a)[2000][100][4] = A; 
  
      // Get global position in X direction 
      int i = get_global_id(0); 
  
      for (j=0;j<100;j++) { 
      for (k=0;k<4;k++) { 
       (*a)[i][j][k] = i*i*i + j*j*j; 
      } 
      } 
  }