UCP/TAG: Fix mt bug in probe - v1.11.x

openucx · Aug 12, 2021 · 7d6fe47 · 7d6fe47
1 parent 06bdf26
commit 7d6fe47
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ucp/tag/probe.c b/src/ucp/tag/probe.c
@@ -28,6 +28,7 @@ UCS_PROFILE_FUNC(ucp_tag_message_h, ucp_tag_probe_nb,
 
     UCP_CONTEXT_CHECK_FEATURE_FLAGS(worker->context, UCP_FEATURE_TAG,
                                     return NULL);
+    UCP_WORKER_THREAD_CS_ENTER_CONDITIONAL(worker);
 
     ucs_trace_req("probe_nb tag %"PRIx64"/%"PRIx64" remove=%d", tag, tag_mask,
                   rem);
@@ -47,6 +48,7 @@ UCS_PROFILE_FUNC(ucp_tag_message_h, ucp_tag_probe_nb,
         }
     }
 
+    UCP_WORKER_THREAD_CS_EXIT_CONDITIONAL(worker);
 
     return rdesc;
 }