@@ -44,6 +44,7 @@ struct urUSMSharedAllocTest
4444
4545 ur_usm_pool_handle_t pool = nullptr ;
4646 bool usePool = std::get<0 >(getParam()).value;
47+ void *ptr = nullptr ;
4748};
4849
4950// The 0 value parameters are not relevant for urUSMSharedAllocTest tests, they
@@ -57,17 +58,16 @@ UUR_TEST_SUITE_P(
5758 uur::printUSMAllocTestString<urUSMSharedAllocTest>);
5859
5960TEST_P (urUSMSharedAllocTest, Success) {
60- void *ptr = nullptr ;
6161 size_t allocation_size = sizeof (int );
6262 ASSERT_SUCCESS (urUSMSharedAlloc (context, device, nullptr , pool,
6363 allocation_size, &ptr));
6464 ASSERT_NE (ptr, nullptr );
6565
6666 ur_event_handle_t event = nullptr ;
6767 uint8_t pattern = 0 ;
68- ASSERT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
68+ EXPECT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
6969 allocation_size, 0 , nullptr , &event));
70- ASSERT_SUCCESS (urEventWait (1 , &event));
70+ EXPECT_SUCCESS (urEventWait (1 , &event));
7171
7272 ASSERT_SUCCESS (urUSMFree (context, ptr));
7373 EXPECT_SUCCESS (urEventRelease (event));
@@ -85,17 +85,16 @@ TEST_P(urUSMSharedAllocTest, SuccessWithDescriptors) {
8585 ur_usm_desc_t usm_desc{UR_STRUCTURE_TYPE_USM_DESC, &usm_host_desc,
8686 /* mem advice flags */ UR_USM_ADVICE_FLAG_DEFAULT,
8787 /* alignment */ 0 };
88- void *ptr = nullptr ;
8988 size_t allocation_size = sizeof (int );
9089 ASSERT_SUCCESS (urUSMSharedAlloc (context, device, &usm_desc, pool,
9190 allocation_size, &ptr));
9291 ASSERT_NE (ptr, nullptr );
9392
9493 ur_event_handle_t event = nullptr ;
9594 uint8_t pattern = 0 ;
96- ASSERT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
95+ EXPECT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
9796 allocation_size, 0 , nullptr , &event));
98- ASSERT_SUCCESS (urEventWait (1 , &event));
97+ EXPECT_SUCCESS (urEventWait (1 , &event));
9998
10099 ASSERT_SUCCESS (urUSMFree (context, ptr));
101100 EXPECT_SUCCESS (urEventRelease (event));
@@ -107,31 +106,28 @@ TEST_P(urUSMSharedAllocTest, SuccessWithMultipleAdvices) {
107106 /* mem advice flags */ UR_USM_ADVICE_FLAG_SET_READ_MOSTLY |
108107 UR_USM_ADVICE_FLAG_BIAS_CACHED,
109108 /* alignment */ 0 };
110- void *ptr = nullptr ;
111109 size_t allocation_size = sizeof (int );
112110 ASSERT_SUCCESS (urUSMSharedAlloc (context, device, &usm_desc, pool,
113111 allocation_size, &ptr));
114112 ASSERT_NE (ptr, nullptr );
115113
116114 ur_event_handle_t event = nullptr ;
117115 uint8_t pattern = 0 ;
118- ASSERT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
116+ EXPECT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
119117 allocation_size, 0 , nullptr , &event));
120- ASSERT_SUCCESS (urEventWait (1 , &event));
118+ EXPECT_SUCCESS (urEventWait (1 , &event));
121119
122120 ASSERT_SUCCESS (urUSMFree (context, ptr));
123121 EXPECT_SUCCESS (urEventRelease (event));
124122}
125123
126124TEST_P (urUSMSharedAllocTest, InvalidNullHandleContext) {
127- void *ptr = nullptr ;
128125 ASSERT_EQ_RESULT (
129126 UR_RESULT_ERROR_INVALID_NULL_HANDLE,
130127 urUSMSharedAlloc (nullptr , device, nullptr , pool, sizeof (int ), &ptr));
131128}
132129
133130TEST_P (urUSMSharedAllocTest, InvalidNullHandleDevice) {
134- void *ptr = nullptr ;
135131 ASSERT_EQ_RESULT (
136132 UR_RESULT_ERROR_INVALID_NULL_HANDLE,
137133 urUSMSharedAlloc (context, nullptr , nullptr , pool, sizeof (int ), &ptr));
@@ -144,14 +140,12 @@ TEST_P(urUSMSharedAllocTest, InvalidNullPtrMem) {
144140}
145141
146142TEST_P (urUSMSharedAllocTest, InvalidUSMSize) {
147- void *ptr = nullptr ;
148143 ASSERT_EQ_RESULT (
149144 UR_RESULT_ERROR_INVALID_USM_SIZE,
150145 urUSMSharedAlloc (context, device, nullptr , pool, -1 , &ptr));
151146}
152147
153148TEST_P (urUSMSharedAllocTest, InvalidValueAlignPowerOfTwo) {
154- void *ptr = nullptr ;
155149 ur_usm_desc_t desc = {};
156150 desc.stype = UR_STRUCTURE_TYPE_USM_DESC;
157151 desc.align = 5 ;
@@ -185,16 +179,15 @@ TEST_P(urUSMSharedAllocAlignmentTest, SuccessAlignedAllocations) {
185179 /* mem advice flags */ UR_USM_ADVICE_FLAG_DEFAULT,
186180 alignment};
187181
188- void *ptr = nullptr ;
189182 ASSERT_SUCCESS (urUSMSharedAlloc (context, device, &usm_desc, pool,
190183 allocation_size, &ptr));
191184 ASSERT_NE (ptr, nullptr );
192185
193186 ur_event_handle_t event = nullptr ;
194187 uint8_t pattern = 0 ;
195- ASSERT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
188+ EXPECT_SUCCESS (urEnqueueUSMFill (queue, ptr, sizeof (pattern), &pattern,
196189 allocation_size, 0 , nullptr , &event));
197- ASSERT_SUCCESS (urEventWait (1 , &event));
190+ EXPECT_SUCCESS (urEventWait (1 , &event));
198191
199192 ASSERT_SUCCESS (urUSMFree (context, ptr));
200193 EXPECT_SUCCESS (urEventRelease (event));
0 commit comments