We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e0f289f commit bb46346Copy full SHA for bb46346
numba_cuda/numba/cuda/tests/cudapy/test_ipc.py
@@ -225,17 +225,12 @@ def test_staged(self):
225
226
# Test on every CUDA devices
227
ngpus = len(cuda.gpus)
228
- futures = [
229
- self.exe.submit(
+
+ for device_num in range(ngpus):
230
+ future = self.exe.submit(
231
staged_ipc_handle_test, ipch, device_num, parent_pid=os.getpid()
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())
+ np.testing.assert_equal(arr, future.result(timeout=FUTURE_TIMEOUT))
239
240
def test_ipc_array(self):
241
for device_num in range(len(cuda.gpus)):
0 commit comments