ep support logprob (#4089) (#4151)

2025-12-24 13:28:13 +08:00 · 2025-09-19 14:07:31 +08:00
parent a685e5ad35
commit 66a98b44ed
3 changed files with 11 additions and 17 deletions
--- a/custom_ops/gpu_ops/get_output_msg_with_topk.cc
+++ b/custom_ops/gpu_ops/get_output_msg_with_topk.cc
@@ -39,9 +39,6 @@ void GetOutputTopK(const paddle::Tensor& x,
                   int k,
                   int64_t rank_id,
                   bool wait_flag) {
-    if (rank_id > 0) {
-        return;
-    }

    static struct msgdata msg_rcv;
    int msg_queue_id = 1;