jax.experimental.pallas.mosaic_gpu.GPUMemorySpace#

class jax.experimental.pallas.mosaic_gpu.GPUMemorySpace(value)[原始碼]#

一個枚舉。

__init__()#

屬性

GMEM

全域記憶體。

SMEM

共享記憶體。

REGS

暫存器。