triton.experimental.gluon.language.nvidia.blackwell.TensorMemoryScalesLayout

class triton.experimental.gluon.language.nvidia.blackwell.TensorMemoryScalesLayout(self, cga_layout: List[List[int]] = <factory>)

Describes the layout for tensor memory scales in Blackwell architecture.

Parameters:

cga_layout (Optional[List[List[int]]]) – CGA layout bases. Defaults to [].

__init__(self, cga_layout: List[List[int]] = <factory>) → None

Methods

__init__(self, cga_layout)
mangle(self)

Attributes

cga_layout