jax.experimental.pallas.mosaic_gpu.Layout

class jax.experimental.pallas.mosaic_gpu.Layout(value)

列舉。

__init__()

屬性

WGMMA	[m, n] 矩陣，其中 m % 64 == 0 == n % 8。
WGMMA_ROW	[m] 矩陣，其中 m % 64 == 0。
WG_SPLAT
WG_STRIDED