Skip to content

Commit 1f90e3f

Browse files
committed
added a dummy config for reduction kernels
1 parent 44e07d7 commit 1f90e3f

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2558,7 +2558,15 @@ def make_config(x, r, num_warps=None, num_stages=1, register_intensive=False):
25582558

25592559
# Additional reduction configs appended for ROCm builds
25602560
if torch.version.hip:
2561-
pass
2561+
# New config
2562+
result_configs.append(triton_config_reduction(
2563+
size_hints,
2564+
8192,
2565+
2048,
2566+
num_warps=4,
2567+
num_stages=1,
2568+
waves_per_eu=2
2569+
))
25622570

25632571
return result_configs
25642572

0 commit comments

Comments
 (0)