ucx: fix latency test

as per openucx/ucx#1492
artpol84 · Jun 29, 2017 · 51e4bff · 51e4bff
1 parent 4f42685
commit 51e4bff
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/ucx/latency/ucp_latency.c b/ucx/latency/ucp_latency.c
@@ -183,6 +183,12 @@ int recv_msg(ucp_worker_h ucp_worker, ucp_ep_h server_ep)
         /* Following blocked methods used to polling internal file descriptor
          * to make CPU idle and don't spin loop
          */
+        /* Progressing before probe to update the state */
+        ucp_worker_progress(ucp_worker);
+
+        /* Probing incoming events in non-block mode */
+        msg_tag = ucp_tag_probe_nb(ucp_worker, tag, tag_mask, 1, &info_tag);
+
         if (ucp_test_mode == TEST_MODE_WAIT) {
             /* Polling incoming events*/
             status = ucp_worker_wait(ucp_worker);
@@ -196,11 +202,6 @@ int recv_msg(ucp_worker_h ucp_worker, ucp_ep_h server_ep)
             }
         }
 
-        /* Progressing before probe to update the state */
-        ucp_worker_progress(ucp_worker);
-
-        /* Probing incoming events in non-block mode */
-        msg_tag = ucp_tag_probe_nb(ucp_worker, tag, tag_mask, 1, &info_tag);
     } while (msg_tag == NULL);
 
     if( same_buf ) {