Better debugging messages

c50f3d6f · Chenggang Zhao · eef7ab50 · c50f3d6f
Commit c50f3d6f authored Jul 08, 2025 by Chenggang Zhao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

csrc/kernels/utils.cuh csrc/kernels/utils.cuh +2 -2

No files found.
--- a/csrc/kernels/utils.cuh
+++ b/csrc/kernels/utils.cuh
@@ -463,8 +463,8 @@ barrier_block(int** barrier_signal_ptrs, int rank) {
        if (__all_sync(0xffffffff, value <= 0))
            break;

-        if (clock64() - start_time > NUM_TIMEOUT_CYCLES and get_lane_id() == 0) {
-            printf("DeepEP timeout check failed: rank = %d, thread = %d)\n", rank, thread_id);
+        if (clock64() - start_time > NUM_TIMEOUT_CYCLES and thread_id < kNumRanks) {
+            printf("DeepEP timeout check failed: rank = %d, thread = %d, value = %d)\n", rank, thread_id, value);
            trap();
        }
    }