cupyx.jit.shared_memory

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

Allocates shared memory and returns it as a 1-D array.

Parameters
  • dtype (dtype) – The dtype of the returned array.

  • size (int or None) – If int type, the size of static shared memory. If None, declares the shared memory with extern specifier.

  • alignment (int or None) – Enforce the alignment via __align__(N).