cupyx.jit.shared_memory

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

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

参数:

• dtype (dtype) – 返回数组的数据类型。

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

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