我正在从这个笔记本学习如何在 GPU 上加速 python 计算,其中一行让我感到困惑:
mandel_kernel[griddim, blockdim](-2.0, 1.0, -1.0, 1.0, d_image, 20)
在这里,mandel_kernel
是一个装饰的 (by cuda.jit
) 函数,griddim
并且blockdim
是长度为 2: 的元griddim=(32,16)
组blockdim=(32,8)
。
这个方括号是python语法的函数名和参数列表的一部分,还是特定于cuda.jit
装饰的东西?