diff --git a/src/PTQ/PTQ.cpp b/src/PTQ/PTQ.cpp index 22c34388ec0cd629d837387927f54745a551f80a..05ab48594709f79099ad643b2960b03c96ef7e75 100644 --- a/src/PTQ/PTQ.cpp +++ b/src/PTQ/PTQ.cpp @@ -278,7 +278,7 @@ std::vector<std::shared_ptr<Node>> retrieveNodeVector(std::shared_ptr<GraphView> scheduler.generateScheduling(); // old way : scheduler.forward(); } - nodeVector = scheduler.getStaticScheduling(); + nodeVector = scheduler.getSequentialStaticScheduling(); fixScheduling(nodeVector); nodeVector = removeMatchingNodes(nodeVector, "Producer"); diff --git a/src/QAT/QAT_FixedQ.cpp b/src/QAT/QAT_FixedQ.cpp index 6ada53239f92d19f96dc87e0b91247aa093caecf..c182d6cc5b8402dabbf33c706ba8f406d4e6a162 100644 --- a/src/QAT/QAT_FixedQ.cpp +++ b/src/QAT/QAT_FixedQ.cpp @@ -154,7 +154,7 @@ void QuantFixedQ::devQAT(std::shared_ptr<GraphView> graphView) { SequentialScheduler scheduler(graphView); scheduler.generateScheduling(); - auto s = scheduler.getStaticScheduling(); + auto s = scheduler.getSequentialStaticScheduling(); for (std::shared_ptr<Node> node : s) Log::info(" name : {} ", node->name()); } diff --git a/unit_tests/Test_QuantPTQ.cpp b/unit_tests/Test_QuantPTQ.cpp index e7211ce4092f789c8c6263671ad236b97934ffbb..018b111bfb8355af7c1cff96a603eafdd1bd6c64 100644 --- a/unit_tests/Test_QuantPTQ.cpp +++ b/unit_tests/Test_QuantPTQ.cpp @@ -200,7 +200,7 @@ TEST_CASE("[tmp] basic test") { // //no need to do this anymore, forward does it autimatically now ... // //scheduler.generateScheduling(true); -// std::vector<std::shared_ptr<Node>> ordered_graph_view = scheduler.getStaticScheduling(); +// std::vector<std::shared_ptr<Node>> ordered_graph_view = scheduler.getSequentialStaticScheduling(); // printf("Going to quantize network :\n"); @@ -226,7 +226,7 @@ TEST_CASE("[tmp] basic test") { // scheduler_v2.forward(); // scheduler_v2.generateScheduling(false); -// std::vector<std::shared_ptr<Node>> ordered_graph_view_v2 = scheduler_v2.getStaticScheduling(); +// std::vector<std::shared_ptr<Node>> ordered_graph_view_v2 = scheduler_v2.getSequentialStaticScheduling(); // if(verbose) { // printf("Ordered graph after quantization :\n");