triton.experimental.gluon.language.DistributedLinearLayout

class triton.experimental.gluon.language.DistributedLinearLayout(self, reg_bases: List[List[int]], lane_bases: List[List[int]], warp_bases: List[List[int]], block_bases: List[List[int]], shape: List[int])

Represents a linear distributed layout with explicit bases at register, lane, warp, and block levels. See: https://arxiv.org/abs/2505.23819 for reference.

Parameters:

• reg_bases (List[List[int]]) – Bases for register-level distribution.

• lane_bases (List[List[int]]) – Bases for lane-level distribution.

• warp_bases (List[List[int]]) – Bases for warp-level distribution.

• block_bases (List[List[int]]) – Bases for block-level distribution.

• shape (List[int]) – The tensor global shape.

__init__(self, reg_bases: List[List[int]], lane_bases: List[List[int]], warp_bases: List[List[int]], block_bases: List[List[int]], shape: List[int]) → None

Methods

__init__(self, reg_bases, lane_bases, ...)
format_hardware_view(self, shape)
format_tensor_view(self, shape)
mangle(self)

Attributes

rank
type
reg_bases
lane_bases
warp_bases
block_bases
shape