Add timeout to child process join in tests.

Author: Andy-Jost

cuda_core/tests/memory_ipc/test_event_ipc.py

@@ -58,7 +58,7 @@ def test_main(self, ipc_device, ipc_memory_resource):
 
         log("releasing stream1")
         latch.release()
-        process.join()
+        process.join(timeout=CHILD_TIMEOUT_SEC)
         assert process.exitcode == 0
         log("done")
 
@@ -150,7 +150,7 @@ def test_main(self, ipc_device, busy_waited_sync, use_options_cls, use_option_kw
         assert props[4] is None
         assert props[5] is None
 
-        process.join()
+        process.join(timeout=CHILD_TIMEOUT_SEC)
         assert process.exitcode == 0
 
     def child_main(self, q_in, q_out):

cuda_core/tests/memory_ipc/test_leaks.py

@@ -36,7 +36,7 @@ def exec_success(obj, number=1):
     for _ in range(number):
         process = mp.Process(target=child_main, args=(obj,))
         process.start()
-        process.join()
+        process.join(timeout=CHILD_TIMEOUT_SEC)
         assert process.exitcode == 0
 
 
@@ -52,7 +52,7 @@ def exec_launch_failure(obj, number=1):
     for _ in range(number):
         process = mp.Process(target=child_main_bad, args=(obj,))
         process.start()
-        process.join()
+        process.join(timeout=CHILD_TIMEOUT_SEC)
         assert process.exitcode != 0
 
 
@@ -135,6 +135,6 @@ def prime():
     if not prime_was_run:
         process = mp.Process()
         process.start()
-        process.join()
+        process.join(timeout=CHILD_TIMEOUT_SEC)
         assert process.exitcode == 0
         prime_was_run = True