triton.experimental.gluon.amd.cdna5.TensorDescriptor

class triton.experimental.gluon.amd.cdna5.TensorDescriptor(base: Any, shape: List[int], strides: List[int], block_shape: List[int], layout: triton.experimental.gluon.language._layouts.PaddedSharedLayout | triton.experimental.gluon.language._layouts.SwizzledSharedLayout | triton.experimental.gluon.language.amd.cdna5._layouts.PartitionedSharedLayout, padding: str = 'zero')

__init__(self, base: Any, shape: List[int], strides: List[int], block_shape: List[int], layout: PaddedSharedLayout | SwizzledSharedLayout | PartitionedSharedLayout, padding: str = 'zero') → None

Methods

__init__(self, base, shape, strides, ...[, ...])
from_tensor(tensor, block_shape, layout)	Create a TensorDescriptor object from a tensor.

Attributes

padding
base
shape
strides
block_shape
layout