triton.language.device_print

triton.language.device_print(prefix, *args)

Print the values at runtime from the device. String formatting does not work for runtime values, so you should provide the values you want to print as arguments. The first value must be a string, all following values must be scalars or tensors.

Calling the Python builtin print is the same as calling this function, and the requirements for the arguments will match this function (not the normal requirements for print).

tl.device_print("pid", pid)
print("pid", pid)
Parameters:
  • prefix – a prefix to print before the values. This is required to be a string literal.

  • args – the values to print. They can be any tensor or scalar.