cupyx.jit.grid

cupyx.jit.grid(ndim) = <cupyx.jit function>

计算网格中的线程索引。

第一个整数的计算方式如下

jit.threadIdx.x + jit.blockIdx.x * jit.blockDim.x

对于另外两个整数，使用 y 和 z 属性。

参数:

ndim (int) – 网格的维度。只允许 1, 2 或 3。

返回:

如果 ndim 是 1，则返回一个整数，否则返回一个元组。

返回类型:

int or tuple

注意

此函数遵循 Numba 的 numba.cuda.grid() 的惯例。