如何在没有分支的情况下实现 2D 环面

Question

我需要为 CUDA 内核实现 2D 圆环，因此如果可能，它应该没有分支。

例如

  |  |  |
--0--1--2--
--3--4--5--
--6--7--8--
  |  |  |

所以线程 0 应该知道他的邻居是 1、3、6 和 2；等等。

我得到左右邻居如下：

right = (id + 1) % columns + firstRowId
left  = (id - 1) % columns + firstRowId

但我不知道如何有效地获取上下 id。

更新：firstRowId 表示其行的第一个元素的 id，不好的名字，对不起。

Answer 1

我认为这些应该可以解决问题：

n = rows * columns;
up = (id - columns + n) % n;
down = (id + columns) % n;