cupyx.jit.shared_memory

cupyx.jit.shared_memory = <cupyx.jit._builtin_funcs.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.

cupyx.jit.syncthreads cupyx.jit._interface._JitRawKernel