triton.experimental.gluon.language.SliceLayout

class triton.experimental.gluon.language.SliceLayout(self, dim: int, parent: DistributedLayout)

Represents a layout corresponding to slicing a distributed tensor along one dimension.

Parameters:

• dim (int) – The dimension index to slice.

• parent (DistributedLayout) – The parent layout before slicing.

__init__(self, dim: int, parent: DistributedLayout) → None

Methods

__init__(self, dim, parent)
format_hardware_view(self, shape)
format_tensor_view(self, shape)
mangle(self)

Attributes

cga_layout
rank
type
dim
parent