Skip to content

Commit a99c369

Browse files
committed
[Tests] fix segfault in multi_device_event_cache_tests
Using event created on a context associated with device1 on a queue associated with a different device was causing as segault on urEnqueueEventsWait
1 parent 187ac04 commit a99c369

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

test/adapters/level_zero/multi_device_event_cache_tests.cpp

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -54,12 +54,12 @@ TEST_F(urMultiQueueMultiDeviceEventCacheTest,
5454
ASSERT_SUCCESS(urDevicePartition(devices[0], &properties, numSubDevices,
5555
sub_devices.data(), nullptr));
5656
uur::raii::Context context1 = nullptr;
57-
ASSERT_SUCCESS(
58-
urContextCreate(1, &sub_devices[0], nullptr, context1.ptr()));
57+
ASSERT_SUCCESS(urContextCreate(sub_devices.size(), &sub_devices[0], nullptr,
58+
context1.ptr()));
5959
ASSERT_NE(nullptr, context1);
6060
uur::raii::Context context2 = nullptr;
61-
ASSERT_SUCCESS(
62-
urContextCreate(1, &sub_devices[1], nullptr, context2.ptr()));
61+
ASSERT_SUCCESS(urContextCreate(sub_devices.size(), &sub_devices[0], nullptr,
62+
context2.ptr()));
6363
ASSERT_NE(nullptr, context2);
6464
ur_queue_handle_t queue1 = nullptr;
6565
ASSERT_SUCCESS(urQueueCreate(context1, sub_devices[0], 0, &queue1));
@@ -91,10 +91,12 @@ TEST_F(urMultiQueueMultiDeviceEventCacheTest,
9191
GTEST_SKIP();
9292
}
9393
uur::raii::Context context1 = nullptr;
94-
ASSERT_SUCCESS(urContextCreate(1, &devices[0], nullptr, context1.ptr()));
94+
ASSERT_SUCCESS(
95+
urContextCreate(devices.size(), &devices[0], nullptr, context1.ptr()));
9596
ASSERT_NE(nullptr, context1);
9697
uur::raii::Context context2 = nullptr;
97-
ASSERT_SUCCESS(urContextCreate(1, &devices[1], nullptr, context2.ptr()));
98+
ASSERT_SUCCESS(
99+
urContextCreate(devices.size(), &devices[0], nullptr, context2.ptr()));
98100
ASSERT_NE(nullptr, context2);
99101
ur_queue_handle_t queue1 = nullptr;
100102
ASSERT_SUCCESS(urQueueCreate(context1, devices[0], 0, &queue1));

0 commit comments

Comments
 (0)