Skip to content

Commit 83e3be0

Browse files
committed
Backport InductorConfig class.
1 parent dbdb554 commit 83e3be0

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,14 @@
6969
)
7070

7171

72+
class InductorConfig(Config):
73+
"""Inductor-specific Triton config with additional control flags"""
74+
75+
def __init__(self, *args, dynamic_scale_rblock=True, **kwargs):
76+
super().__init__(*args, **kwargs)
77+
self.dynamic_scale_rblock = dynamic_scale_rblock
78+
79+
7280
class NoTritonConfigsError(RuntimeError):
7381
pass
7482

0 commit comments

Comments
 (0)