triton.language.sort

triton.language.sort(ptr, dim=-1, descending=False, _builder=None)

Triton sort 前端接口

参数：

ptr: tl.tensor，输入张量 dim: int 或 tl.constexpr[int]，排序维度 descending: bool 或 tl.constexpr[bool]，是否降序 _builder: ir.builder，底层 IR 构建器

返回：

values: tl.tensor，排序后的值（类型与输入一致）