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

[diffusion] fix: fix bug of copy_if (#20094)

Browse Source 
Co-authored-by: Yihan Chen <yingluosanqian@gmail.com>
This commit is contained in:
Fan Lin
2026-03-08 14:27:58 +08:00
committed by GitHub
co-authored by Yihan Chen
parent 7f9f85d4c8
commit 7fb282a96f
1 changed files with 1 additions and 1 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
+1 -1
View File
@@ -162,7 +162,7 @@ class ScaleResidualNormScaleShift:
@cute.jit @cute.jit
def copy_if(src, dst): def copy_if(src, dst):
if cutlass.const_expr( if cutlass.const_expr(
isinstance(src, cute.Tensor) and isinstance(src, cute.Tensor) isinstance(src, cute.Tensor) and isinstance(dst, cute.Tensor)
): ):
cute.autovec_copy(src, dst) # LDG.128 cute.autovec_copy(src, dst) # LDG.128