Conditionalize xblock values for HIP.

naromero77amd · naromero77amd · commit f80f0b8f826a · 2025-11-14T23:04:04.000Z
diff --git a/torch/_inductor/runtime/triton_heuristics.py b/torch/_inductor/runtime/triton_heuristics.py
@@ -2097,9 +2097,14 @@ def _persistent_reduction_configs(
         or inductor_meta.get("max_autotune_pointwise")
     )
 
+    if torch.version.hip:
+        xblock_vals = [1, 4, 8, 16, 32, 64, 128, 256]
+    else:
+        xblock_vals = [1, 8, 32, 128]
+
     configs = [
         triton_config_reduction(size_hints, xblock, rnumel, register_intensive=True)
-        for xblock in (1, 8, 32, 128)
+        for xblock in xblock_vals
         if xblock == 1 or (xblock <= xnumel and (max_autotune_enabled or rnumel * xblock <= 4096))
     ]
 

Original file line number	Diff line number	Diff line change
`@@ -2097,9 +2097,14 @@ def _persistent_reduction_configs(`
`2097`	`2097`	`or inductor_meta.get("max_autotune_pointwise")`
`2098`	`2098`	`)`
`2099`	`2099`
	`2100`	`+ if torch.version.hip:`
	`2101`	`+ xblock_vals = [1, 4, 8, 16, 32, 64, 128, 256]`
	`2102`	`+ else:`
	`2103`	`+ xblock_vals = [1, 8, 32, 128]`
	`2104`	`+`
`2100`	`2105`	`configs = [`
`2101`	`2106`	`triton_config_reduction(size_hints, xblock, rnumel, register_intensive=True)`
`2102`		`- for xblock in (1, 8, 32, 128)`
	`2107`	`+ for xblock in xblock_vals`
`2103`	`2108`	`if xblock == 1 or (xblock <= xnumel and (max_autotune_enabled or rnumel * xblock <= 4096))`
`2104`	`2109`	`]`
`2105`	`2110`