使用 CUDA 优化向量元素交换

Question

因为我是cuda新手..我需要你的帮助 我有这个长向量，对于每组 24 个元素，我需要执行以下操作： 对于前 12 个元素，偶数元素乘以 -1， 对于第二个 12 个元素，奇数元素乘以 -1，然后进行以下交换：

图表：因为我还没有足够的点，所以无法发布图像，所以这里是：

http://www.freeimagehosting.net/image.php?e4b88fb666.png

我写过这段代码，想知道您是否可以帮助我进一步优化它以解决分歧或银行冲突..

//subvector is a multiple of 24, Mds and Nds are shared memory

____shared____ double Mds[subVector];

____shared____ double Nds[subVector];

int tx = threadIdx.x;
int tx_mod = tx ^ 0x0001;
int  basex = __umul24(blockDim.x, blockIdx.x);

 Mds[tx] = M.elements[basex + tx];
__syncthreads();

// flip the signs 
 if (tx < (tx/24)*24 + 12)
 {  
    //if < 12 and even
    if ((tx & 0x0001)==0)
    Mds[tx] = -Mds[tx];
 }
 else
 if (tx < (tx/24)*24 + 24)
 {
    //if >12 and < 24 and odd
    if ((tx & 0x0001)==1)
    Mds[tx] = -Mds[tx];
 }

 __syncthreads();

 if (tx < (tx/24)*24 + 6)
 {  
//for the first 6 elements .. swap with last six in the 24elements group (see graph)
    Nds[tx] = Mds[tx_mod + 18];
    Mds [tx_mod + 18] = Mds [tx];
    Mds[tx] = Nds[tx];
 }
 else
 if (tx < (tx/24)*24 + 12)
 {
    // for the second 6 elements .. swp with next adjacent group (see graph)
    Nds[tx] = Mds[tx_mod + 6];
    Mds [tx_mod + 6] = Mds [tx];
    Mds[tx] = Nds[tx];
 }   
__syncthreads();

提前致谢..

Answer 1

保罗在之前的问题中为您提供了很好的起点。

需要注意的几件事：您正在进行非基数 2 除法，这是昂贵的。
相反，尝试利用线程块的多维性质。例如，将 x 维度设置为 24，这样就无需进行除法。

一般来说，尝试调整线程块尺寸以反映您的数据尺寸。

简化符号翻转：例如，如果你不想翻转符号，你仍然可以乘以恒等号1。弄清楚如何仅使用算术将偶数/奇数映射到 1 和 -1：例如 sign = (even*2+1) - 2，其中 Even 是 1 或 0。