stopping server after test

dmitrygx · Dec 28, 2018 · e827c4b · e827c4b
1 parent 105476e
commit e827c4b
Show file tree

Hide file tree

Showing 13 changed files with 21 additions and 4 deletions.
diff --git a/benchmarks/bibw-asyncio.py b/benchmarks/bibw-asyncio.py
@@ -73,6 +73,7 @@ async def talk_to_client(client_ep):
         recv_buffer_region.free_host()
 
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/benchmarks/bibw-future.py b/benchmarks/bibw-future.py
@@ -80,6 +80,7 @@ def talk_to_client(client_ep):
 
     ucp.destroy_ep(client_ep)
     cb_not_done = False
+    ucp.stop_server()
 
 def talk_to_server(ip, port):
 

diff --git a/benchmarks/lat-asyncio-blind-recv.py b/benchmarks/lat-asyncio-blind-recv.py
@@ -55,6 +55,7 @@ async def talk_to_client(client_ep):
         send_buffer_region.free_host()
 
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/benchmarks/lat-asyncio-fast.py b/benchmarks/lat-asyncio-fast.py
@@ -59,6 +59,7 @@ async def talk_to_client(client_ep):
         recv_buffer_region.free_host()
 
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/benchmarks/lat-asyncio.py b/benchmarks/lat-asyncio.py
@@ -59,6 +59,7 @@ async def talk_to_client(client_ep):
         recv_buffer_region.free_host()
 
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/benchmarks/lat-fast.py b/benchmarks/lat-fast.py
@@ -94,6 +94,7 @@ def talk_to_client(client_ep):
 
     ucp.destroy_ep(client_ep)
     cb_not_done = False
+    ucp.stop_server()
 
 def talk_to_server(ip, port):
 

diff --git a/benchmarks/unibw-asyncio.py b/benchmarks/unibw-asyncio.py
@@ -62,6 +62,7 @@ async def talk_to_client(client_ep):
         send_buffer_region.free_host()
 
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/benchmarks/unibw-future.py b/benchmarks/unibw-future.py
@@ -70,6 +70,7 @@ def talk_to_client(client_ep):
 
     ucp.destroy_ep(client_ep)
     cb_not_done = False
+    ucp.stop_server()
 
 def talk_to_server(ip, port):
 

diff --git a/tests/test-duality-data-validity-msg-preallocated.py b/tests/test-duality-data-validity-msg-preallocated.py
@@ -39,6 +39,7 @@ async def talk_to_client(client_ep):
     send_buffer_region.free_cuda()
     recv_buffer_region.free_cuda()
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/tests/test-duality-data-validity.py b/tests/test-duality-data-validity.py
@@ -33,6 +33,7 @@ async def talk_to_client(client_ep):
 
     buffer_region.free_cuda()
     ucp.destroy_ep(client_ep)
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/tests/test-duality.py b/tests/test-duality.py
@@ -27,9 +27,12 @@ async def talk_to_client(client_ep):
 
     buffer_region.free_cuda()
     ucp.destroy_ep(client_ep)
+    print("done with talk_to_client")
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 
+    print("in talk_to_server")
     msg_log = max_msg_log
 
     server_ep = ucp.get_endpoint(ip, port)
@@ -45,6 +48,7 @@ async def talk_to_server(ip, port):
 
     buffer_region.free_cuda()
     ucp.destroy_ep(server_ep)
+    print("done with talk_to_server")
 
 parser = argparse.ArgumentParser()
 parser.add_argument('-s','--server', help='enter server ip', required=False)
@@ -74,9 +78,10 @@ async def talk_to_server(ip, port):
     asyncio.gather(coro_server, coro_client)
 )
 
-try:
-    loop.run_forever()
-except KeyboardInterrupt:
-    pass
+#try:
+#    loop.run_forever()
+#except KeyboardInterrupt:
+#    pass
 
 loop.close()
+ucp.fin()
diff --git a/tests/test-server-listen-accept-future-await.py b/tests/test-server-listen-accept-future-await.py
@@ -34,6 +34,7 @@ async def talk_to_client(client_ep):
     recv_buffer_region.free_cuda()
     ucp.destroy_ep(client_ep)
     print('talk_to_client done')
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):
 

diff --git a/tests/test-server-listen-accept-future-recv.py b/tests/test-server-listen-accept-future-recv.py
@@ -29,6 +29,7 @@ async def talk_to_client(client_ep):
     ucp.destroy_ep(client_ep)
 
     print("passed talk_to_client")
+    ucp.stop_server()
 
 async def talk_to_server(ip, port):