Update internode_ll.cu (#246)

fbcf4300 · fzyzcjy · GitHub · c95997f8 · fbcf4300
Unverified Commit fbcf4300 authored Jun 23, 2025 by fzyzcjy Committed by GitHub Jun 23, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/kernels/internode_ll.cu csrc/kernels/internode_ll.cu +1 -1

No files found.
--- a/csrc/kernels/internode_ll.cu
+++ b/csrc/kernels/internode_ll.cu
@@ -498,7 +498,7 @@ combine(void* combined_x,
    }
    cg::this_grid().sync();

-    // Reduce tokens with FP8 cast
+    // Reduce tokens
    EP_DEVICE_ASSERT(num_topk <= 32 and hidden_bf16_int4 <= num_threads);
    EP_STATIC_ASSERT(kHidden % (32 * kNumElemsPerInt4) == 0, "Invalid vectorization");
    if (thread_id < hidden_bf16_int4) {