@@ -165,6 +165,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, WriteRead) {
165
165
ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[1 ], buffer, false , 0 ,
166
166
buffer_size_bytes, out_vec.data (), 1 ,
167
167
&e1 , nullptr ));
168
+
169
+ ASSERT_SUCCESS (urQueueFinish (queues[1 ]));
170
+
168
171
for (auto &a : out_vec) {
169
172
ASSERT_EQ (a, fill_val);
170
173
}
@@ -186,6 +189,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, FillRead) {
186
189
ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[1 ], buffer, false , 0 ,
187
190
buffer_size_bytes, out_vec.data (), 1 ,
188
191
&e1 , nullptr ));
192
+
193
+ ASSERT_SUCCESS (urQueueFinish (queues[1 ]));
194
+
189
195
for (auto &a : out_vec) {
190
196
ASSERT_EQ (a, fill_val);
191
197
}
@@ -219,6 +225,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, WriteKernelRead) {
219
225
ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[0 ], buffer, false , 0 ,
220
226
buffer_size_bytes, out_vec.data (), 1 ,
221
227
&e2 , nullptr ));
228
+
229
+ ASSERT_SUCCESS (urQueueFinish (queues[0 ]));
230
+
222
231
for (auto &a : out_vec) {
223
232
ASSERT_EQ (a, fill_val + 1 );
224
233
}
@@ -257,6 +266,9 @@ TEST_F(urMultiDeviceContextMemBufferTest, WriteKernelKernelRead) {
257
266
ASSERT_SUCCESS (urEnqueueMemBufferRead (queues[1 ], buffer, false , 0 ,
258
267
buffer_size_bytes, out_vec.data (), 1 ,
259
268
&e3 , nullptr ));
269
+
270
+ ASSERT_SUCCESS (urQueueFinish (queues[1 ]));
271
+
260
272
for (auto &a : out_vec) {
261
273
ASSERT_EQ (a, fill_val + 2 );
262
274
}
0 commit comments