cupyx.jit.gridsize

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

计算网格大小。

第一个整数的计算如下:

jit.blockDim.x * jit.gridDim.x

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

参数:

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

返回:

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

返回类型:

int or tuple

备注

此函数遵循 Numba 的 numba.cuda.gridsize() 约定。