From aa92042794a621013579f19ad5fc9ae07957d524 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9B=A8=E6=9D=AD?=
 <11237004+y0hang@user.noreply.gitee.com>
Date: Thu, 22 May 2025 03:21:24 +0000
Subject: [PATCH] [Huawei] Enhanced the MPI log locating capability.

Offering: Kunpeng HPC
---
 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 d467f7d6cb..14b76b04a7 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 85c64fdd45..ef2c01d960 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