优化combine

b1ba831f · zhanghj2 · 91691124 · b1ba831f
Commit b1ba831f authored Feb 06, 2026 by zhanghj2
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/smxx/decode/combine/combine.cu csrc/smxx/decode/combine/combine.cu +1 -1

No files found.
--- a/csrc/smxx/decode/combine/combine.cu
+++ b/csrc/smxx/decode/combine/combine.cu
@@ -191,7 +191,7 @@ flash_fwd_mla_combine_kernel(const CombineParams params) {
                //     return res;
                // };

-                // float4 data = result[i];
+                float4 data = result[i];
                // o_ptr[i * 4] = float32_to_bfloat16((data.x));
                // o_ptr[i * 4 + 1] = float32_to_bfloat16((data.y));
                // o_ptr[i * 4 + 2] = float32_to_bfloat16((data.z));