cupyx.jit.shared_memory

cupyx.jit.shared_memory = <cupyx.jit._compile.SharedMemory object>

Allocates shared memory and returns the 1-dim 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.