From e178ca3a844747fd083d3a6a14f02f0dab3a2aa1 Mon Sep 17 00:00:00 2001 From: x30073543 Date: Fri, 26 Sep 2025 14:50:40 +0800 Subject: [PATCH] fix(autotune): delete default tune args from all_kwargs --- .../python/triton_patch/runtime/autotiling_tuner.py | 11 +---------- triton_patch/python/triton_patch/runtime/autotuner.py | 7 +++---- 2 files changed, 4 insertions(+), 14 deletions(-) diff --git a/triton_patch/python/triton_patch/runtime/autotiling_tuner.py b/triton_patch/python/triton_patch/runtime/autotiling_tuner.py index 2ca12a5..357b450 100644 --- a/triton_patch/python/triton_patch/runtime/autotiling_tuner.py +++ b/triton_patch/python/triton_patch/runtime/autotiling_tuner.py @@ -122,16 +122,7 @@ class AutoTilingTuner(Autotuner): if len(self.gen_configs) == 0 and len(self.user_configs) == 0: return [ - Config( - {}, - num_warps=4, - num_stages=2, - num_ctas=1, - num_buffers_warp_spec=0, - num_consumer_groups=0, - reg_dec_producer=0, - reg_inc_consumer=0, - ) + Config({}) ] else: return self.gen_configs + self.user_configs diff --git a/triton_patch/python/triton_patch/runtime/autotuner.py b/triton_patch/python/triton_patch/runtime/autotuner.py index 6c51929..67753e1 100644 --- a/triton_patch/python/triton_patch/runtime/autotuner.py +++ b/triton_patch/python/triton_patch/runtime/autotuner.py @@ -38,8 +38,7 @@ class Autotuner(KernelInterface): """ if not configs: self.configs = [ - Config({}, num_warps=4, num_stages=2, num_ctas=1, num_buffers_warp_spec=0, num_consumer_groups=0, - reg_dec_producer=0, reg_inc_consumer=0) + Config({}) ] else: self.configs = configs @@ -303,8 +302,8 @@ class Config: :ivar bishengir_options: dict of options that pass to bishengir. """ - def __init__(self, kwargs, num_warps=4, num_stages=2, num_ctas=1, num_buffers_warp_spec=0, num_consumer_groups=0, - reg_dec_producer=0, reg_inc_consumer=0, maxnreg=None, pre_hook=None, **bishengir_options): + def __init__(self, kwargs, num_warps=None, num_stages=None, num_ctas=None, num_buffers_warp_spec=None, num_consumer_groups=None, + reg_dec_producer=None, reg_inc_consumer=None, maxnreg=None, pre_hook=None, **bishengir_options): self.kwargs = kwargs self.num_warps = num_warps self.num_ctas = num_ctas -- Gitee