diff --git a/python/triton/runtime/autotuner.py b/python/triton/runtime/autotuner.py index 8115776beb65..73e61866273c 100644 --- a/python/triton/runtime/autotuner.py +++ b/python/triton/runtime/autotuner.py @@ -239,15 +239,16 @@ def __init__(self, kwargs, num_warps=4, num_stages=2, num_ctas=1, maxnreg=None, self.pre_hook = pre_hook def all_kwargs(self): - return self.kwargs | { - k: v - for (k, v) in ( - ("num_warps", self.num_warps), - ("num_ctas", self.num_ctas), - ("num_stages", self.num_stages), - ("maxnreg", self.maxnreg), - ) - if v is not None + return { + **self.kwargs, **{ + k: v + for (k, v) in ( + ("num_warps", self.num_warps), + ("num_ctas", self.num_ctas), + ("num_stages", self.num_stages), + ("maxnreg", self.maxnreg), + ) if v is not None + } } def __str__(self):