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() 约定。