From 01d88a5c045ed30622aef5a4aeb6d7f0a88ea181 Mon Sep 17 00:00:00 2001 From: Rohan Potdar Date: Fri, 17 Jul 2026 16:29:01 +0000 Subject: [PATCH] [FlyDSL][AOT] Fix cktile_epilogue_silu AOT arg mismatch The silu path in `_precompile_epilogue_to_cache` still passes a runtime `swiglu_limit` (`float("inf")`) to `launch_silu_and_mul_fq`, but that launcher no longer takes a runtime swiglu_limit arg -- swiglu_limit is now a compile-time parameter of `build_silu_and_mul_fq_module`. The extra positional arg makes FlyDSL AOT fail with "too many positional arguments" for every `cktile_epilogue_silu` config, aborting the aiter wheel build (PREBUILD_KERNELS=1). Drop the stale `float("inf")` so the launch tuple matches the launcher (swiglu_limit stays compile-time; default 0.0 = no clamp for silu). Verified: the 12 configs that fail in CI (inter_dim 256..1536, topk 6..9, gfx950) go from 12 failed -> 12 ok under COMPILE_ONLY with flydsl 0.2.2. Signed-off-by: Rohan Potdar Co-Authored-By: Claude Opus 4.8 (1M context) --- aiter/aot/flydsl/moe.py | 1 - 1 file changed, 1 deletion(-) diff --git a/aiter/aot/flydsl/moe.py b/aiter/aot/flydsl/moe.py index 83b17a2593..302e07b1f6 100644 --- a/aiter/aot/flydsl/moe.py +++ b/aiter/aot/flydsl/moe.py @@ -843,7 +843,6 @@ def _precompile_epilogue_to_cache(act: str, inter_dim: int, topk: int): _ptr_view_safe(empty_f32), rows, sorted_token_ids.shape[0], - float("inf"), # swiglu_limit (unused for silu) 0, # stream: null/default (compile-only, kernel is never launched) ), )