triton.max

triton.max(input, axis, builder=None)

Returns the maximum value of all elements in the input block along the provided axis

Parameters

• input – the input values

• axis – the dimension along which the reduction should be done

• builder (triton.ir.builder, optional from within JIT'ed functions) – IR builder to generate code into