Skip to content

Commit e14516a

Browse files
authored
[Docs] Document the TRITON_PRINT_AUTOTUNING flag added in #3411. (#3542)
1 parent 6eadbce commit e14516a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/triton/runtime/autotuner.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,11 @@ def kernel(x_ptr, x_size, **META):
251251
This means that whatever value the kernel updates will be updated multiple times.
252252
To avoid this undesired behavior, you can use the `reset_to_zero` argument, which
253253
resets the value of the provided tensor to `zero` before running any configuration.
254+
255+
If the environment variable :code:`TRITON_PRINT_AUTOTUNING` is set to
256+
:code:`"1"`, Triton will print a message to stdout after autotuning each
257+
kernel, including the time spent autotuning and the best configuration.
258+
254259
:param configs: a list of :code:`triton.Config` objects
255260
:type configs: list[triton.Config]
256261
:param key: a list of argument names whose change in value will trigger the evaluation of all provided configs.

0 commit comments

Comments
 (0)