@@ -108,8 +108,8 @@ TEST_P(memspaceGetTest, providerFromMemspace) {
108
108
umf_memory_provider_handle_t hProvider = nullptr ;
109
109
umf_result_t ret =
110
110
umfMemoryProviderCreateFromMemspace (hMemspace, nullptr , &hProvider);
111
- UT_ASSERTeq (ret, UMF_RESULT_SUCCESS);
112
- UT_ASSERTne (hProvider, nullptr );
111
+ ASSERT_EQ (ret, UMF_RESULT_SUCCESS);
112
+ ASSERT_NE (hProvider, nullptr );
113
113
114
114
umfMemoryProviderDestroy (hProvider);
115
115
}
@@ -120,15 +120,15 @@ TEST_P(memspaceProviderTest, allocFree) {
120
120
size_t alignment = 0 ;
121
121
122
122
umf_result_t ret = umfMemoryProviderAlloc (hProvider, size, alignment, &ptr);
123
- UT_ASSERTeq (ret, UMF_RESULT_SUCCESS);
124
- UT_ASSERTne (ptr, nullptr );
123
+ ASSERT_EQ (ret, UMF_RESULT_SUCCESS);
124
+ ASSERT_NE (ptr, nullptr );
125
125
126
126
// Access the allocation, so that all the pages associated with it are
127
127
// allocated on some NUMA node.
128
128
memset (ptr, 0xFF , size);
129
129
130
130
ret = umfMemoryProviderFree (hProvider, ptr, size);
131
- UT_ASSERTeq (ret, UMF_RESULT_SUCCESS);
131
+ ASSERT_EQ (ret, UMF_RESULT_SUCCESS);
132
132
}
133
133
134
134
static std::vector<int > getAllCpus () {
@@ -148,20 +148,20 @@ TEST_P(memspaceProviderTest, allocLocalMt) {
148
148
auto pinAllocValidate = [&](umf_memory_provider_handle_t hProvider,
149
149
int cpu) {
150
150
hwloc_topology_t topology = NULL ;
151
- UT_ASSERTeq (hwloc_topology_init (&topology), 0 );
152
- UT_ASSERTeq (hwloc_topology_load (topology), 0 );
151
+ ASSERT_EQ (hwloc_topology_init (&topology), 0 );
152
+ ASSERT_EQ (hwloc_topology_load (topology), 0 );
153
153
154
154
// Pin current thread to the provided CPU.
155
155
hwloc_cpuset_t pinCpuset = hwloc_bitmap_alloc ();
156
- UT_ASSERTeq (hwloc_bitmap_set (pinCpuset, cpu), 0 );
157
- UT_ASSERTeq (
158
- hwloc_set_cpubind (topology, pinCpuset, HWLOC_CPUBIND_THREAD), 0 );
156
+ ASSERT_EQ (hwloc_bitmap_set (pinCpuset, cpu), 0 );
157
+ ASSERT_EQ ( hwloc_set_cpubind (topology, pinCpuset, HWLOC_CPUBIND_THREAD),
158
+ 0 );
159
159
160
160
// Confirm that the thread is pinned to the provided CPU.
161
161
hwloc_cpuset_t curCpuset = hwloc_bitmap_alloc ();
162
- UT_ASSERTeq (
163
- hwloc_get_cpubind (topology, curCpuset, HWLOC_CPUBIND_THREAD), 0 );
164
- UT_ASSERT (hwloc_bitmap_isequal (curCpuset, pinCpuset));
162
+ ASSERT_EQ ( hwloc_get_cpubind (topology, curCpuset, HWLOC_CPUBIND_THREAD),
163
+ 0 );
164
+ ASSERT_TRUE (hwloc_bitmap_isequal (curCpuset, pinCpuset));
165
165
hwloc_bitmap_free (curCpuset);
166
166
hwloc_bitmap_free (pinCpuset);
167
167
@@ -172,8 +172,8 @@ TEST_P(memspaceProviderTest, allocLocalMt) {
172
172
173
173
umf_result_t ret =
174
174
umfMemoryProviderAlloc (hProvider, size, alignment, &ptr);
175
- UT_ASSERTeq (ret, UMF_RESULT_SUCCESS);
176
- UT_ASSERTne (ptr, nullptr );
175
+ ASSERT_EQ (ret, UMF_RESULT_SUCCESS);
176
+ ASSERT_NE (ptr, nullptr );
177
177
178
178
// Access the allocation, so that all the pages associated with it are
179
179
// allocated on some NUMA node.
@@ -194,20 +194,20 @@ TEST_P(memspaceProviderTest, allocLocalMt) {
194
194
hwloc_location loc;
195
195
loc.location .object = allocNodeObj,
196
196
loc.type = hwloc_location_type_alias::HWLOC_LOCATION_TYPE_OBJECT;
197
- UT_ASSERTeq (hwloc_get_local_numanode_objs (topology, &loc, &nNodes,
198
- localNodes.data (), 0 ),
199
- 0 );
200
- UT_ASSERT (nNodes <= MAX_NODES);
197
+ ASSERT_EQ (hwloc_get_local_numanode_objs (topology, &loc, &nNodes,
198
+ localNodes.data (), 0 ),
199
+ 0 );
200
+ ASSERT_LE (nNodes, MAX_NODES);
201
201
202
202
// Confirm that the allocation from this thread was made to a local
203
203
// NUMA node.
204
- UT_ASSERT (std::any_of (localNodes.begin (), localNodes.end (),
205
- [&allocNodeObj](hwloc_obj_t node) {
206
- return node == allocNodeObj;
207
- }));
204
+ ASSERT_TRUE (std::any_of (localNodes.begin (), localNodes.end (),
205
+ [&allocNodeObj](hwloc_obj_t node) {
206
+ return node == allocNodeObj;
207
+ }));
208
208
209
209
ret = umfMemoryProviderFree (hProvider, ptr, size);
210
- UT_ASSERTeq (ret, UMF_RESULT_SUCCESS);
210
+ ASSERT_EQ (ret, UMF_RESULT_SUCCESS);
211
211
212
212
hwloc_topology_destroy (topology);
213
213
};
0 commit comments