triton.experimental.gluon.language.amd.cdna5.tdm.make_tensor_descriptor
- triton.experimental.gluon.language.amd.cdna5.tdm.make_tensor_descriptor(base: tensor, shape: List[constexpr | tensor], strides: List[constexpr | tensor], block_shape: List[constexpr], layout: PaddedSharedLayout | SwizzledSharedLayout | PartitionedSharedLayout, _semantic=None) tensor_descriptor
Make a tensor descriptor object.
AMD CDNA5 TDM tensor descriptors only support zero padding.
- Parameters:
base (tensor) – base pointer of the tensor in global memory.
shape (List[int]) – shape of the tensor.
strides (List[int]) – strides of the tensor.
block_shape (List[int]) – block shape of the tensor.
layout (PaddedSharedLayout | SwizzledSharedLayout | PartitionedSharedLayout) – the layout of the tensor in shared memory.
- Returns:
the created tensor descriptor object
- Return type: