jax.experimental.pallas.mosaic_gpu.GPUMemorySpace#

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

一个枚举。

__init__()#

属性

GMEM

全局内存。

SMEM

共享内存。

TMEM

张量内存。

REGS

寄存器。