From 1ac5c7288d6714845d28a58aa39d1aab0b7cd377 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 12 Jan 2026 11:59:18 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- unsloth/kernels/moe/autotune_cache.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/unsloth/kernels/moe/autotune_cache.py b/unsloth/kernels/moe/autotune_cache.py index f11748aa73..91740d2a9d 100644 --- a/unsloth/kernels/moe/autotune_cache.py +++ b/unsloth/kernels/moe/autotune_cache.py @@ -156,7 +156,9 @@ def get_or_autotune_moe_kernels( # This avoids the 2-4 minute startup cost. if not force_autotune: # We can try to rely on heuristic - logger.info(f"Using Heuristic (Safe) MoE kernel configs for SM{device_capability[0]}{device_capability[1]}") + logger.info( + f"Using Heuristic (Safe) MoE kernel configs for SM{device_capability[0]}{device_capability[1]}" + ) return _get_heuristic_configs() if not force_autotune and cache_key in _kernel_config_cache: logger.info(f"Using in-memory cached MoE kernel configs: {cache_key}") @@ -375,7 +377,6 @@ def _run_moe_autotuning( return config_fwd, config_bwd_dx, config_bwd_dw - return config_fwd, config_bwd_dx, config_bwd_dw @@ -400,7 +401,7 @@ def _get_heuristic_configs() -> Tuple[Any, Any, Any]: permute_x = True, permute_y = True, use_tma_load_x = False, - use_tma_load_w = False, # TMA loads might need alignment checks, safer to disable for heuristic + use_tma_load_w = False, # TMA loads might need alignment checks, safer to disable for heuristic use_tma_store = False, )