cupyx.jit.shared_memory

cupyx.jit.shared_memory(dtype, size, alignment=None) = <cupyx.jit function>

分配共享内存并将其作为一维数组返回。

参数:

• dtype (dtype) – 返回数组的dtype。

• size (int or None) – 如果是 int 类型，则表示静态共享内存的大小。如果是 None，则使用 extern 说明符声明共享内存。

• alignment (int or None) – 通过 __align__(N) 强制对齐。