diff --git a/pybind/myucp.c b/pybind/myucp.c index fac2c8e823b..2cca43e8c79 100644 --- a/pybind/myucp.c +++ b/pybind/myucp.c @@ -245,7 +245,6 @@ static unsigned ucp_ipy_worker_progress(ucp_worker_h ucp_worker) num_cb_free++; assert(num_cb_free <= CB_Q_MAX_ENTRIES); assert(cb_free_head.tqh_first != NULL); - printf("calling pyx_cb\b"); tmp_pyx_cb(tmp_arg, tmp_py_cb); } return status; diff --git a/tests/test-server-listen-accept-future-recv.py b/tests/test-server-listen-accept-future-recv.py index c30a240935c..d4e74b960b1 100644 --- a/tests/test-server-listen-accept-future-recv.py +++ b/tests/test-server-listen-accept-future-recv.py @@ -47,35 +47,38 @@ def send_recv(ep, msg_log, server, is_cuda): async def talk_to_client(client_ep): print("in talk_to_client") + msg_log = max_msg_log - ''' buffer_region = ucp.buffer_region() buffer_region.alloc_cuda(1 << msg_log) - msg.set_mem(0, 1 << msg_log) - #await ep.send(msg, 1 << msg_log) - msg = ucp.ucp_msg(buffer_region) + + send_req = await client_ep.send(msg, 1 << msg_log) + + recv_req = await client_ep.recv_ft() + buffer_region.free_cuda() - ''' print(42) return 42 async def talk_to_server(ip, port): + msg_log = max_msg_log + server_ep = ucp.get_endpoint(ip, port) - ''' buffer_region = ucp.buffer_region() buffer_region.alloc_cuda(1 << msg_log) - msg.set_mem(0, 1 << msg_log) - #await ep.send(msg, 1 << msg_log) - msg = ucp.ucp_msg(buffer_region) + + recv_req = await server_ep.recv_ft() + + send_req = await server_ep.send(msg, 1 << msg_log) + buffer_region.free_cuda() - ''' print(3.14) return 3.14