Skip to content

Commit 9ba040c

Browse files
faresscfaressc
committed
Adapted the tests for the limited number of threads on the github runners
1 parent c8f70a2 commit 9ba040c

File tree

2 files changed

+201
-48
lines changed

2 files changed

+201
-48
lines changed

‎test/scheduler/test_InferenceManager.cpp‎

Lines changed: 103 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ TEST_P(InferenceManagerTest, Simple){
4040

4141
PrePostProcessor pp_processor(test_params.inference_config);
4242
BackendBase* custom_processor = nullptr; // Use default processor
43-
ContextConfig context_config;
43+
ContextConfig context_config(2); // Use 2 threads for testing, gh runner max on macOS is 3
4444

4545
InferenceManager inference_manager(
4646
pp_processor,
@@ -65,7 +65,7 @@ TEST_P(InferenceManagerTest, WithCustomLatency) {
6565

6666
PrePostProcessor pp_processor(test_params.inference_config);
6767
BackendBase* custom_processor = nullptr; // Use default processor
68-
ContextConfig context_config;
68+
ContextConfig context_config(2); // Use 2 threads for testing, gh runner max on macOS is 3
6969

7070
InferenceManager inference_manager(
7171
pp_processor,
@@ -102,7 +102,7 @@ TEST_P(InferenceManagerTest, WithEmptyCustomLatency) {
102102

103103
PrePostProcessor pp_processor(test_params.inference_config);
104104
BackendBase* custom_processor = nullptr; // Use default processor
105-
ContextConfig context_config;
105+
ContextConfig context_config(2); // Use 2 threads for testing, gh runner max on macOS is 3
106106

107107
InferenceManager inference_manager(
108108
pp_processor,
@@ -140,7 +140,7 @@ TEST_P(InferenceManagerTest, WithPartialCustomLatency) {
140140

141141
PrePostProcessor pp_processor(test_params.inference_config);
142142
BackendBase* custom_processor = nullptr; // Use default processor
143-
ContextConfig context_config;
143+
ContextConfig context_config(2); // Use 2 threads for testing, gh runner max on macOS is 3
144144

145145
InferenceManager inference_manager(
146146
pp_processor,
@@ -217,7 +217,11 @@ INSTANTIATE_TEST_SUITE_P(
217217
InferenceConfig(
218218
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
219219
std::vector<TensorShape>{TensorShape({{1, 1, 2048}}, {{1, 1, 2048}})},
220-
1.f
220+
1.f,
221+
0,
222+
false,
223+
0.f,
224+
2
221225
),
222226
{ 4095 }
223227
},
@@ -272,7 +276,8 @@ INSTANTIATE_TEST_SUITE_P(
272276
4.f,
273277
0,
274278
false,
275-
0.f
279+
0.f,
280+
2
276281
),
277282
{ 12 }
278283
},
@@ -281,7 +286,11 @@ INSTANTIATE_TEST_SUITE_P(
281286
InferenceConfig(
282287
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
283288
std::vector<TensorShape>{TensorShape({{1, 1, 1}}, {{1, 1, 2048}})},
284-
40.f
289+
40.f,
290+
0,
291+
false,
292+
0.f,
293+
2
285294
),
286295
{ 5885 }
287296
},
@@ -290,7 +299,11 @@ INSTANTIATE_TEST_SUITE_P(
290299
InferenceConfig(
291300
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
292301
std::vector<TensorShape>{TensorShape({{1, 1, 2048}}, {{1, 1, 1}})},
293-
40.f
302+
40.f,
303+
0,
304+
false,
305+
0.f,
306+
2
294307
),
295308
{ 1 }
296309
},
@@ -299,7 +312,11 @@ INSTANTIATE_TEST_SUITE_P(
299312
InferenceConfig(
300313
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
301314
std::vector<TensorShape>{TensorShape({{1, 1, 2048}}, {{1, 1, 1}})},
302-
39.f
315+
39.f,
316+
0,
317+
false,
318+
0.f,
319+
2
303320
),
304321
{ 1 }
305322
},
@@ -308,7 +325,11 @@ INSTANTIATE_TEST_SUITE_P(
308325
InferenceConfig(
309326
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
310327
std::vector<TensorShape>{TensorShape({{1, 1, 1}}, {{1, 1, 2048}})},
311-
50.f
328+
50.f,
329+
0,
330+
false,
331+
0.f,
332+
2
312333
),
313334
{ 4096 },
314335
},
@@ -318,7 +339,11 @@ INSTANTIATE_TEST_SUITE_P(
318339
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
319340
std::vector<TensorShape>{TensorShape({{1, 1, 2048}}, {{1, 4, 1}})},
320341
ProcessingSpec({1}, {4}),
321-
20.f
342+
20.f,
343+
0,
344+
false,
345+
0.f,
346+
2
322347
),
323348
{ 1 }
324349
},
@@ -328,7 +353,11 @@ INSTANTIATE_TEST_SUITE_P(
328353
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
329354
std::vector<TensorShape>{TensorShape({{1, 4, 1}}, {{1, 1, 2048}})},
330355
ProcessingSpec({4}, {1}),
331-
40.f
356+
40.f,
357+
0,
358+
false,
359+
0.f,
360+
2
332361
),
333362
{ 3960 }
334363
},
@@ -338,7 +367,11 @@ INSTANTIATE_TEST_SUITE_P(
338367
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
339368
std::vector<TensorShape>{TensorShape({{1, 16, 1}}, {{1, 1, 2048}})},
340369
ProcessingSpec({16}, {1}),
341-
40.f
370+
40.f,
371+
0,
372+
false,
373+
0.f,
374+
2
342375
),
343376
{ 2048 }
344377
},
@@ -348,7 +381,11 @@ INSTANTIATE_TEST_SUITE_P(
348381
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
349382
std::vector<TensorShape>{TensorShape({{1, 16, 1}}, {{1, 1, 2048}, {2, 256}})},
350383
ProcessingSpec({16}, {1, 2}),
351-
40.f
384+
40.f,
385+
0,
386+
false,
387+
0.f,
388+
2
352389
),
353390
{ 6144, 768 }
354391
},
@@ -358,7 +395,11 @@ INSTANTIATE_TEST_SUITE_P(
358395
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
359396
std::vector<TensorShape>{TensorShape({{1, 16, 1}, {2, 256}}, {{1, 1, 2048}, {3, 128}})},
360397
ProcessingSpec({16, 2}, {1, 3}),
361-
40.f
398+
40.f,
399+
0,
400+
false,
401+
0.f,
402+
2
362403
),
363404
{ 2048, 128 }
364405
},
@@ -368,7 +409,11 @@ INSTANTIATE_TEST_SUITE_P(
368409
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
369410
std::vector<TensorShape>{TensorShape({{1, 16, 1}, {2, 256}}, {{1, 1, 2048}, {3, 128}})},
370411
ProcessingSpec({16, 2}, {1, 3}),
371-
40.f
412+
40.f,
413+
0,
414+
false,
415+
0.f,
416+
2
372417
),
373418
{ 6144, 384 }
374419
},
@@ -378,7 +423,11 @@ INSTANTIATE_TEST_SUITE_P(
378423
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
379424
std::vector<TensorShape>{TensorShape({{1, 16, 1}, {2, 256}}, {{1, 2, 2048}})},
380425
ProcessingSpec({16, 2}, {2}),
381-
40.f
426+
40.f,
427+
0,
428+
false,
429+
0.f,
430+
2
382431
),
383432
{ 4924 }
384433
},
@@ -388,7 +437,11 @@ INSTANTIATE_TEST_SUITE_P(
388437
InferenceConfig(
389438
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
390439
std::vector<TensorShape>{TensorShape({{1, 1, 2048}}, {{1, 1, 2048}})},
391-
13.f
440+
13.f,
441+
0,
442+
false,
443+
0.f,
444+
2
392445
),
393446
{ 2744 }
394447
},
@@ -397,7 +450,11 @@ INSTANTIATE_TEST_SUITE_P(
397450
InferenceConfig(
398451
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
399452
std::vector<TensorShape>{TensorShape({{1, 1, 1024}}, {{1, 1, 1024}})},
400-
40.f
453+
40.f,
454+
0,
455+
false,
456+
0.f,
457+
2
401458
),
402459
{ 2820 }
403460
},
@@ -407,7 +464,11 @@ INSTANTIATE_TEST_SUITE_P(
407464
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
408465
std::vector<TensorShape>{TensorShape({{1, 8, 1}}, {{1, 1, 1024}})},
409466
ProcessingSpec({8}, {1}),
410-
12.f
467+
12.f,
468+
0,
469+
false,
470+
0.f,
471+
2
411472
),
412473
{ 3072 }
413474
},
@@ -417,7 +478,11 @@ INSTANTIATE_TEST_SUITE_P(
417478
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
418479
std::vector<TensorShape>{TensorShape({{1, 8, 1}}, {{1, 1, 1024}})},
419480
ProcessingSpec({8}, {1}),
420-
13.f
481+
13.f,
482+
0,
483+
false,
484+
0.f,
485+
2
421486
),
422487
{ 3583 }
423488
},
@@ -427,27 +492,39 @@ INSTANTIATE_TEST_SUITE_P(
427492
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
428493
std::vector<TensorShape>{TensorShape({{1, 4, 1}, {2, 128}}, {{1, 1, 512}, {3, 64}})},
429494
ProcessingSpec({4, 2}, {1, 1}),
430-
50.f
495+
50.f,
496+
0,
497+
false,
498+
0.f,
499+
2
431500
),
432-
{ 6656, 2496 }
501+
{ 12800, 4800 }
433502
},
434503
InferenceManagerTestParams {
435504
HostConfig(1500, 44100./8., true, 1),
436505
InferenceConfig(
437506
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
438507
std::vector<TensorShape>{TensorShape({{1, 4, 1}, {2, 128}, {1, 2}}, {{1, 1, 512}, {3, 64}})},
439508
ProcessingSpec({4, 2, 1}, {1, 1}),
440-
50.f
509+
50.f,
510+
0,
511+
false,
512+
0.f,
513+
2
441514
),
442-
{ 6656, 2496 }
515+
{ 12800, 4800 }
443516
},
444517
InferenceManagerTestParams {
445518
HostConfig(256., 48000./8, false, 1),
446519
InferenceConfig(
447520
std::vector<ModelData>{ModelData("placeholder", anira::InferenceBackend::CUSTOM)},
448521
std::vector<TensorShape>{TensorShape({{1, 4, 1}, {2, 256}}, {{1, 1, 2048}, {3, 128}})},
449522
ProcessingSpec({1, 2}, {1, 1}, {0, 256}, {2048, 0}),
450-
40.f
523+
40.f,
524+
0,
525+
false,
526+
0.f,
527+
2
451528
),
452529
{ 2048, 0 }
453530
}

0 commit comments

Comments
 (0)