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

Co-authored-by: Yihan Chen <yingluosanqian@gmail.com>
2026-03-08 14:27:58 +08:00
parent 7f9f85d4c8
commit 7fb282a96f
1 changed files with 1 additions and 1 deletions
--- 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
@@ -162,7 +162,7 @@ class ScaleResidualNormScaleShift:
        @cute.jit
        def copy_if(src, dst):
            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