diff --git a/recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py b/recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py index 10ecd74ce9..5131068b99 100644 --- a/recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py +++ b/recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py @@ -92,7 +92,7 @@ def add_fn(x, y): # # .. note:: # -# ``torch.compile`` only supports configs and key arguments to ``triton.autotune``. +# ``torch.compile`` only supports configs, keys, restore_value, and reset_to_zero arguments to ``triton.autotune``. if not has_triton(): print("Skipping because triton is not supported on this device.")