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