Merge pull request #7231 from brminich/ucp/fix_probe_bug_v.1.11

UCP/TAG: Fix mt bug in probe - v1.11.x
openucx · Aug 13, 2021 · f08684a · f08684a
2 parents 06bdf26 + 7d6fe47
commit f08684a
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;
 }