triton.language.cat

triton.language.cat(input, other, can_reorder=False)

Concatenate the given blocks

Parameters:

• input (Tensor) – The first input tensor.

• other (Tensor) – The second input tensor.

• reorder – Compiler hint. If true, the compiler is allowed to reorder elements while concatenating inputs. Only use if the order does not matter (e.g., result is only used in reduction ops). Current implementation of cat supports only can_reorder=True.