print userdata

jeffnvidia · Nov 17, 2024 · 6387df8 · 6387df8
1 parent e377e85
commit 6387df8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/components/tl/ucp/tl_ucp_context.c b/src/components/tl/ucp/tl_ucp_context.c
@@ -137,12 +137,12 @@ static int memcpy_device_start(void *dest, void *src, size_t size,
                                 void *completion, void *user_data) {
 
         ucc_info("using memcpy_device_start");
-        printf("printf : using memcpy_device_start");
         ucc_status_t status;
         ucc_ee_executor_task_args_t eargs;
         ucc_ee_executor_t *exec;
         ucc_tl_ucp_task_t *task = (ucc_tl_ucp_task_t *) user_data;
 
+        ucc_info("memcpystart user data : %p", user_data);
         status = ucc_coll_task_get_executor(&task->super, &exec);
         if (ucc_unlikely(status != UCC_OK)) {
             return status;
@@ -175,13 +175,13 @@ static int memcpy_device_start(void *dest, void *src, size_t size,
 static int memcpy_device(void *dest, void *src, size_t size, void *user_data){
 
     ucc_info("using memcpy_device");
-    printf("printf : using memcpy_device");
     ucc_status_t status;
     ucc_ee_executor_task_args_t eargs;
     ucc_ee_executor_t *exec;
     ucc_ee_executor_task_t *etask;
     ucc_tl_ucp_task_t *task = (ucc_tl_ucp_task_t *) user_data;
 
+    ucc_info("memcpy user data : %p", user_data);
     status = ucc_coll_task_get_executor(&task->super, &exec);
     if (ucc_unlikely(status != UCC_OK)) {
         return status;

diff --git a/src/components/tl/ucp/tl_ucp_sendrecv.h b/src/components/tl/ucp/tl_ucp_sendrecv.h
@@ -121,6 +121,7 @@ ucc_tl_ucp_send_common_with_mem(void *buffer, size_t msglen, ucc_memory_type_t m
     req_param.user_data   = user_data;
     req_param.memh       = mh;
     task->tagged.send_posted++;
+    ucc_info("before ucx, user_data : %p", user_data);
     return ucp_tag_send_nbx(ep, buffer, 1, ucp_tag, &req_param);
 }
 
@@ -230,6 +231,7 @@ ucc_tl_ucp_recv_common_with_mem(void *buffer, size_t msglen, ucc_memory_type_t m
     req_param.user_data   = user_data;
     req_param.memh = mh;
     task->tagged.recv_posted++;
+    ucc_info("before ucx (recv) : userdata : %p", user_data);
     return ucp_tag_recv_nbx(team->worker->ucp_worker, buffer, 1, ucp_tag,
                             ucp_tag_mask, &req_param);
 }