From 2ca9a1934912de40fc82828bc2256cc6203af8e8 Mon Sep 17 00:00:00 2001
From: Post-Dawn <635426116@qq.com>
Date: Thu, 22 May 2025 16:28:09 +0800
Subject: [PATCH] [Huawei] Enhanced the MPI log locating capability.

---
 ompi/mca/pml/ucx/pml_ucx.c         | 4 ++++
 ompi/mca/pml/ucx/pml_ucx_request.c | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/ompi/mca/pml/ucx/pml_ucx.c b/ompi/mca/pml/ucx/pml_ucx.c
index d467f7d6cb3..14b76b04a7d 100644
--- a/ompi/mca/pml/ucx/pml_ucx.c
+++ b/ompi/mca/pml/ucx/pml_ucx.c
@@ -427,6 +427,10 @@ static ucp_ep_h mca_pml_ucx_add_proc_common(ompi_proc_t *proc)
     ep_params.field_mask = UCP_EP_PARAM_FIELD_REMOTE_ADDRESS;
     ep_params.address    = address;
 
+    ep_params.field_mask   |= UCP_EP_PARAM_FIELD_PEER_HOST_ADDR;
+    ep_params.vpid          = proc->super.proc_name.vpid;
+    ep_params.peer_hostname = proc->super.proc_hostname;
+
     status = ucp_ep_create(ompi_pml_ucx.ucp_worker, &ep_params, &ep);
     free(address);
     if (UCS_OK != status) {
diff --git a/ompi/mca/pml/ucx/pml_ucx_request.c b/ompi/mca/pml/ucx/pml_ucx_request.c
index 85c64fdd459..ef2c01d9600 100644
--- a/ompi/mca/pml/ucx/pml_ucx_request.c
+++ b/ompi/mca/pml/ucx/pml_ucx_request.c
@@ -211,7 +211,7 @@ void mca_pml_ucx_request_timeout_warn(uint64_t tag)
     if (rc != OPAL_SUCCESS) {
         peer_hostname = "unknown";
     }
-    PML_UCX_WARN("UCP request timeout! request tag 0x%lX local proc: %u peer proc: %d peer hostname: %s\n",
+    PML_UCX_WARN("UCP recv request timeout! request tag 0x%lX local proc: %u peer proc: %d peer hostname: %s\n",
                   tag, OMPI_PROC_MY_NAME->vpid, rank, peer_hostname);
 }
 
-- 
Gitee