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