处理lse为正无穷的情况

csrc/smxx/decode/combine/combine.cu

@@ -95,11 +95,14 @@ flash_fwd_mla_combine_kernel(const CombineParams params) {
                 o_scale = 0.0f;
             }
             else
+            {
+                if (!flash::is_positive_infinity(global_lse))
                 {
                     float Attn_sink_exp2 = __builtin_amdgcn_exp2f(attn_sink * CUDART_L2E_F);
                     float lse_exp2 = __builtin_amdgcn_exp2f(global_lse * CUDART_L2E_F);
                     o_scale = lse_exp2 / (lse_exp2 + Attn_sink_exp2);
                 }
+            }
 
 
             // if (global_lse != INFINITY) {