@@ -165,6 +165,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, WriteRead) {
165165 ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[1 ], buffer, false , 0 ,
166166 buffer_size_bytes, out_vec.data (), 1 ,
167167 &e1 , nullptr ));
168+
169+ ASSERT_SUCCESS (urQueueFinish (queues[1 ]));
170+
168171 for (auto &a : out_vec) {
169172 ASSERT_EQ (a, fill_val);
170173 }
@@ -186,6 +189,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, FillRead) {
186189 ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[1 ], buffer, false , 0 ,
187190 buffer_size_bytes, out_vec.data (), 1 ,
188191 &e1 , nullptr ));
192+
193+ ASSERT_SUCCESS (urQueueFinish (queues[1 ]));
194+
189195 for (auto &a : out_vec) {
190196 ASSERT_EQ (a, fill_val);
191197 }
@@ -219,6 +225,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, WriteKernelRead) {
219225 ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[0 ], buffer, false , 0 ,
220226 buffer_size_bytes, out_vec.data (), 1 ,
221227 &e2 , nullptr ));
228+
229+ ASSERT_SUCCESS (urQueueFinish (queues[0 ]));
230+
222231 for (auto &a : out_vec) {
223232 ASSERT_EQ (a, fill_val + 1 );
224233 }
@@ -257,6 +266,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, WriteKernelKernelRead) {
257266 ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[1 ], buffer, false , 0 ,
258267 buffer_size_bytes, out_vec.data (), 1 ,
259268 &e3 , nullptr ));
269+
270+ ASSERT_SUCCESS (urQueueFinish (queues[1 ]));
271+
260272 for (auto &a : out_vec) {
261273 ASSERT_EQ (a, fill_val + 2 );
262274 }
0 commit comments