minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[SGLang-Diffusion] Fix custom op fake impl missing eps default for torch.compile (#19725)

Browse Source 
Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
This commit is contained in:
赵晨阳
2026-03-03 15:24:36 +08:00
committed by GitHub
co-authored by Xiaoyu Zhang
parent 3c01b44700
commit 62480ebb1b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py
+2 -2
View File
@@ -341,7 +341,7 @@ def fused_norm_scale_shift(
@fused_norm_scale_shift.register_fake @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) y = x.new_empty(x.shape)
return y return y
@@ -424,7 +424,7 @@ def fused_scale_residual_norm_scale_shift(
@fused_scale_residual_norm_scale_shift.register_fake @fused_scale_residual_norm_scale_shift.register_fake
def _fused_scale_residual_norm_scale_shift_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) y = x.new_empty(x.shape)
residual_out = x.new_empty(x.shape) residual_out = x.new_empty(x.shape)