diff --git a/python/sglang/multimodal_gen/runtime/models/dits/minimax_h3.py b/python/sglang/multimodal_gen/runtime/models/dits/minimax_h3.py index 2dc5f3b67..fef29927a 100644 --- a/python/sglang/multimodal_gen/runtime/models/dits/minimax_h3.py +++ b/python/sglang/multimodal_gen/runtime/models/dits/minimax_h3.py @@ -440,6 +440,7 @@ def _apply_qk_norm( and q.stride(-2) == k.stride(-2) == head_dim and q_norm.eps == k_norm.eps and not torch.compiler.is_compiling() + and current_platform.is_cuda() ): fused_inplace_qknorm( q,