@@ -42,7 +42,14 @@ std::vector<int> get_available_cpus() {
42
42
CPU_ZERO (mask);
43
43
44
44
int ret = sched_getaffinity (0 , sizeof (cpu_set_t ), mask);
45
- UT_ASSERTeq (ret, 0 );
45
+
46
+ if (ret != 0 ) {
47
+ available_cpus.emplace_back (-__LINE__);
48
+ CPU_FREE (mask);
49
+
50
+ return available_cpus;
51
+ }
52
+
46
53
// Get all available cpus.
47
54
printf (" All CPUs: " );
48
55
for (size_t i = 0 ; i < CPU_SETSIZE; ++i) {
@@ -88,13 +95,21 @@ struct testNuma : testing::Test {
88
95
ASSERT_NE (os_memory_provider, nullptr );
89
96
}
90
97
91
- struct bitmask *retrieve_nodemask (void *addr) {
98
+ std::pair< int , bitmask *> retrieve_nodemask (void *addr) {
92
99
struct bitmask *retrieved_nodemask = numa_allocate_nodemask ();
93
- UT_ASSERTne (nodemask, nullptr );
100
+
101
+ if (nodemask == nullptr ){
102
+ return std::make_pair (__LINE__, nodemask);
103
+ }
104
+
94
105
int ret = get_mempolicy (nullptr , retrieved_nodemask->maskp ,
95
106
nodemask->size , addr, MPOL_F_ADDR);
96
- UT_ASSERTeq (ret, 0 );
97
- return retrieved_nodemask;
107
+
108
+ if (ret != 0 ) {
109
+ return std::make_pair (__LINE__, retrieved_nodemask);
110
+ }
111
+
112
+ return std::pair (0 , retrieved_nodemask);
98
113
}
99
114
100
115
void TearDown () override {
@@ -241,7 +256,18 @@ TEST_P(testNumaOnEachNode, checkModeInterleaveSingleNode) {
241
256
EXPECT_NODE_EQ (ptr, numa_node_number);
242
257
}
243
258
244
- struct testNumaOnEachCpu : testNuma, testing::WithParamInterface<int > {};
259
+ struct testNumaOnEachCpu : testNuma, testing::WithParamInterface<int > {
260
+ void SetUp () override {
261
+ ::testNuma::SetUp ();
262
+
263
+ int cpuNumber = this ->GetParam ();
264
+
265
+ if (cpuNumber < 0 ) {
266
+ GTEST_FAIL () << " ret is not equal to 0 in " << __FILE__ << " : " << -cpuNumber;
267
+ }
268
+ }
269
+
270
+ };
245
271
246
272
INSTANTIATE_TEST_SUITE_P (testNumaNodesAllocationsAllCpus, testNumaOnEachCpu,
247
273
::testing::ValuesIn (get_available_cpus()));
@@ -260,7 +286,7 @@ TEST_P(testNumaOnEachCpu, checkModePreferredEmptyNodeset) {
260
286
int ret = sched_setaffinity (0 , sizeof (cpu_set_t ), mask);
261
287
CPU_FREE (mask);
262
288
263
- UT_ASSERTeq (ret, 0 );
289
+ ASSERT_EQ (ret, 0 );
264
290
265
291
umf_os_memory_provider_params_t os_memory_provider_params =
266
292
UMF_OS_MEMORY_PROVIDER_PARAMS_TEST;
@@ -275,7 +301,8 @@ TEST_P(testNumaOnEachCpu, checkModePreferredEmptyNodeset) {
275
301
276
302
// Verify we're on the expected CPU
277
303
int cpu_check = sched_getcpu ();
278
- UT_ASSERTeq (cpu, cpu_check);
304
+ ASSERT_EQ (cpu, cpu_check);
305
+
279
306
280
307
int numa_node_number = numa_node_of_cpu (cpu);
281
308
printf (" Got CPU: %d, got numa node: %d\n " , cpu, numa_node_number);
@@ -297,7 +324,7 @@ TEST_P(testNumaOnEachCpu, checkModeLocal) {
297
324
int ret = sched_setaffinity (0 , sizeof (cpu_set_t ), mask);
298
325
CPU_FREE (mask);
299
326
300
- UT_ASSERTeq (ret, 0 );
327
+ ASSERT_EQ (ret, 0 );
301
328
302
329
umf_os_memory_provider_params_t os_memory_provider_params =
303
330
UMF_OS_MEMORY_PROVIDER_PARAMS_TEST;
@@ -312,7 +339,7 @@ TEST_P(testNumaOnEachCpu, checkModeLocal) {
312
339
313
340
// Verify we're on the expected CPU
314
341
int cpu_check = sched_getcpu ();
315
- UT_ASSERTeq (cpu, cpu_check);
342
+ ASSERT_EQ (cpu, cpu_check);
316
343
317
344
int numa_node_number = numa_node_of_cpu (cpu);
318
345
printf (" Got CPU: %d, got numa node: %d\n " , cpu, numa_node_number);
@@ -391,7 +418,16 @@ TEST_F(testNuma, checkModeInterleave) {
391
418
EXPECT_NODE_EQ ((char *)ptr + page_size * i, numa_nodes[index ]);
392
419
}
393
420
394
- bitmask *retrieved_nodemask = retrieve_nodemask (ptr);
421
+ // bitmask *retrieved_nodemask = retrieve_nodemask(ptr);
422
+ auto [fileLine, retrieved_nodemask] = retrieve_nodemask (ptr);
423
+ if (fileLine != 0 ) {
424
+ if (retrieved_nodemask == nullptr ) {
425
+ GTEST_FAIL () << " retrieved_nodemask is nullptr " << __FILE__ << " : " << fileLine;
426
+ }
427
+ else {
428
+ GTEST_FAIL () << " ret is not equal to 0 " << __FILE__ << " : " << fileLine;
429
+ }
430
+ }
395
431
int ret = numa_bitmask_equal (retrieved_nodemask, nodemask);
396
432
numa_bitmask_free (retrieved_nodemask);
397
433
0 commit comments