jax.experimental.pallas.mosaic_gpu.GPUMemorySpace#

class jax.experimental.pallas.mosaic_gpu.GPUMemorySpace(value)[source]#

An enumeration.

__init__()#

Attributes

GMEM

Global memory.

SMEM

Shared memory.

REGS

Registers.