Skip to content

Commit bb46346

Browse files
authored
test: revert back to ipc futures that await each iteration (#564)
1 parent e0f289f commit bb46346

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

numba_cuda/numba/cuda/tests/cudapy/test_ipc.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -225,17 +225,12 @@ def test_staged(self):
225225

226226
# Test on every CUDA devices
227227
ngpus = len(cuda.gpus)
228-
futures = [
229-
self.exe.submit(
228+
229+
for device_num in range(ngpus):
230+
future = self.exe.submit(
230231
staged_ipc_handle_test, ipch, device_num, parent_pid=os.getpid()
231232
)
232-
for device_num in range(ngpus)
233-
]
234-
235-
for fut in concurrent.futures.as_completed(
236-
futures, timeout=FUTURE_TIMEOUT * ngpus
237-
):
238-
np.testing.assert_equal(arr, fut.result())
233+
np.testing.assert_equal(arr, future.result(timeout=FUTURE_TIMEOUT))
239234

240235
def test_ipc_array(self):
241236
for device_num in range(len(cuda.gpus)):

0 commit comments

Comments
 (0)