triton.language.cat

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

Concatenate the given blocks

Parameters:
  • input – The first input tensor.

  • other – 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)