triton.language.sort

triton.language.sort(x, dim: constexpr | None = None, descending: constexpr = constexpr[0])

This function can also be called as a member function on tensor, as x.sort(...) instead of sort(x, ...).