Update outdated note in Triton torch.compile tutorial (#3844)

catswe · web-flow · commit 49c2c81ca425 · 2026-04-29T11:30:36.000-07:00
See https://github.com/pytorch/pytorch/blob/dc3fb57650de3ed9e1eced72452a19f136df3cce/torch/_higher_order_ops/triton_kernel_wrap.py#L1843 cc @williamwen42 @msaroufim @anijain2305 @angelayi
diff --git 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.")

Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,7 @@ def add_fn(x, y):`
`92`	`92`	`#`
`93`	`93`	`# .. note::`
`94`	`94`	`#`
`95`		-# ``torch.compile`` only supports configs and key arguments to ``triton.autotune``.
	`95`	+# ``torch.compile`` only supports configs, keys, restore_value, and reset_to_zero arguments to ``triton.autotune``.
`96`	`96`
`97`	`97`	`if not has_triton():`
`98`	`98`	`print("Skipping because triton is not supported on this device.")`