From 62480ebb1ba4ffc065a0d0df7fbf920c7e42243f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B5=B5=E6=99=A8=E9=98=B3?= Date: Mon, 2 Mar 2026 23:24:36 -0800 Subject: [PATCH] [SGLang-Diffusion] Fix custom op fake impl missing eps default for torch.compile (#19725) Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> --- .../diffusion/cutedsl/scale_residual_norm_scale_shift.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py b/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py index ba8782d3d..734b9bbf7 100644 --- a/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py +++ b/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py @@ -341,7 +341,7 @@ def fused_norm_scale_shift( @fused_norm_scale_shift.register_fake -def _fused_norm_scale_shift_fake(x, weight, bias, scale, shift, norm_type, eps): +def _fused_norm_scale_shift_fake(x, weight, bias, scale, shift, norm_type, eps=1e-5): y = x.new_empty(x.shape) return y @@ -424,7 +424,7 @@ def fused_scale_residual_norm_scale_shift( @fused_scale_residual_norm_scale_shift.register_fake def _fused_scale_residual_norm_scale_shift_fake( - residual, x, gate, weight, bias, scale, shift, norm_type, eps + residual, x, gate, weight, bias, scale, shift, norm_type, eps=1e-5 ): y = x.new_empty(x.shape) residual_out = x.new_empty(x.shape)