Merge pull request #7228 from brminich/ucp/probe_bug_fix

UCP/TAG: Fix mt bug in probe
openucx · Aug 13, 2021 · 5279d97 · 5279d97
2 parents c9d0b6e + c20df2b
commit 5279d97
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;
 }