@@ -54,12 +54,12 @@ TEST_F(urMultiQueueMultiDeviceEventCacheTest,
54
54
ASSERT_SUCCESS (urDevicePartition (devices[0 ], &properties, numSubDevices,
55
55
sub_devices.data (), nullptr ));
56
56
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 ()));
59
59
ASSERT_NE (nullptr , context1);
60
60
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 ()));
63
63
ASSERT_NE (nullptr , context2);
64
64
ur_queue_handle_t queue1 = nullptr ;
65
65
ASSERT_SUCCESS (urQueueCreate (context1, sub_devices[0 ], 0 , &queue1));
@@ -91,10 +91,12 @@ TEST_F(urMultiQueueMultiDeviceEventCacheTest,
91
91
GTEST_SKIP ();
92
92
}
93
93
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 ()));
95
96
ASSERT_NE (nullptr , context1);
96
97
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 ()));
98
100
ASSERT_NE (nullptr , context2);
99
101
ur_queue_handle_t queue1 = nullptr ;
100
102
ASSERT_SUCCESS (urQueueCreate (context1, devices[0 ], 0 , &queue1));
0 commit comments