jax.experimental.pallas.mosaic_gpu.Layout

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

一个枚举。

__init__()

属性

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