diff --git a/aidge_backend_cpu/unit_tests/test_scheduler.py b/aidge_backend_cpu/unit_tests/test_scheduler.py index 494f34565ffd644971c97e9adfa06709dee9e36d..b60ff3f01307f22bd8bf635df2d776cb1267d0f5 100644 --- a/aidge_backend_cpu/unit_tests/test_scheduler.py +++ b/aidge_backend_cpu/unit_tests/test_scheduler.py @@ -57,9 +57,9 @@ class test_scheduler(unittest.TestCase): scheduler = aidge_core.SequentialScheduler(graph_view) scheduler.generate_scheduling() - self.assertEqual(len(scheduler.get_static_scheduling()), 10) + self.assertEqual(len(scheduler.get_sequential_static_scheduling()), 10) # Do not care about the order of execution of the producers - self.assertListEqual([i.name() for i in scheduler.get_static_scheduling()[-3:]], EXPECTED_SCHEDULE) + self.assertListEqual([i.name() for i in scheduler.get_sequential_static_scheduling()[-3:]], EXPECTED_SCHEDULE) def test_parallel_scheduling(self): @@ -83,9 +83,9 @@ class test_scheduler(unittest.TestCase): scheduler = aidge_core.SequentialScheduler(graph_view) scheduler.generate_scheduling() - self.assertEqual(len(scheduler.get_static_scheduling()), 11) + self.assertEqual(len(scheduler.get_sequential_static_scheduling()), 11) # Do not care about the order of execution of the producers - self.assertTrue([i.name() for i in scheduler.get_static_scheduling()[-4:]] in EXPECTED_SCHEDULE) + self.assertTrue([i.name() for i in scheduler.get_sequential_static_scheduling()[-4:]] in EXPECTED_SCHEDULE) if __name__ == '__main__': unittest.main() diff --git a/unit_tests/operator/Test_MetaOperator.cpp b/unit_tests/operator/Test_MetaOperator.cpp index adc548b9148ce8431d51f514ac82808accd6017d..bb9027d3ec0d8eb05dbd555f69bd6e8f5147246b 100644 --- a/unit_tests/operator/Test_MetaOperator.cpp +++ b/unit_tests/operator/Test_MetaOperator.cpp @@ -279,9 +279,9 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { REQUIRE(op->getNbConsumedData(1).data == 32768); REQUIRE(op->getNbProducedData(0).data == 34816); REQUIRE(op->getNbProducedData(1).data == 34816); - REQUIRE(microGraphScheduler->getStaticScheduling(0).size() == 26); - REQUIRE(microGraphScheduler->getStaticScheduling(1).size() == 24); - REQUIRE(microGraphScheduler->getStaticScheduling(15).size() == 24); + REQUIRE(microGraphScheduler->getSequentialStaticScheduling(0).size() == 26); + REQUIRE(microGraphScheduler->getSequentialStaticScheduling(1).size() == 24); + REQUIRE(microGraphScheduler->getSequentialStaticScheduling(15).size() == 24); } SECTION("LSTM(forward_values)") {