diff --git a/triton_patch/python/triton_patch/runtime/autotiling_tuner.py b/triton_patch/python/triton_patch/runtime/autotiling_tuner.py index 2ca12a57a1aa6e511bf61ac25f79cfe853e7e781..357b4509ab8aed5de8d7ae8b1bcbd725815cedd3 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 6c519294834d35df927de385f3f0a7b9450bc51f..67753e129eda03a46de2e15cf34105f1e1c294c8 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