@@ -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