From 7fb282a96f92c5d140d36b2bd30848dbd5cb3e8c Mon Sep 17 00:00:00 2001 From: Fan Lin <49047353+linfann@users.noreply.github.com> Date: Sun, 8 Mar 2026 14:27:58 +0800 Subject: [PATCH] [diffusion] fix: fix bug of copy_if (#20094) Co-authored-by: Yihan Chen --- .../diffusion/cutedsl/scale_residual_norm_scale_shift.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 734b9bbf7..792c272d7 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 @@ -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