diff --git a/cuda_core/tests/memory_ipc/test_peer_access.py b/cuda_core/tests/memory_ipc/test_peer_access.py index 5a06133c9b..3f3ca021f7 100644 --- a/cuda_core/tests/memory_ipc/test_peer_access.py +++ b/cuda_core/tests/memory_ipc/test_peer_access.py @@ -94,6 +94,8 @@ def child_main(self, mr, buffer): # Test 1: Buffer accessible from resident device (dev1) - should always work dev1 = Device(1) dev1.set_current() + # Sync dev1 to ensure IPC import operations are complete + dev1.sync() PatternGen(dev1, NBYTES).verify_buffer(buffer, seed=False) # Test 2: Buffer NOT accessible from dev0 initially (peer access not preserved) @@ -106,6 +108,9 @@ def child_main(self, mr, buffer): dev1.set_current() mr.peer_accessible_by = [0] assert mr.peer_accessible_by == (0,) + # Sync dev1 to ensure peer access setup and any pending operations are complete + # before dev0 accesses the peer memory + dev1.sync() dev0.set_current() PatternGen(dev0, NBYTES).verify_buffer(buffer, seed=False)