Skip to content

Commit f5f2377

Browse files
igchornrspruit
authored andcommitted
[CTS] Add missing urQueueFinish to memory-migrate test
All enqueus are done asynchronously so we need to wait for the last operation before checking the values.
1 parent 7583fd2 commit f5f2377

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

test/conformance/memory-migrate/urMemBufferMigrateAcrossDevices.cpp

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)