diff --git a/src/PTQ/PTQ.cpp b/src/PTQ/PTQ.cpp index 156a45d2cfdabd8f302f60ca794c381f69c27d5a..c07816d720025b22a2f6ff6aa16db9f4aa8b5391 100644 --- a/src/PTQ/PTQ.cpp +++ b/src/PTQ/PTQ.cpp @@ -333,16 +333,7 @@ static std::shared_ptr<Tensor> getBiasTensor(std::shared_ptr<Node> node) std::vector<std::shared_ptr<Node>> retrieveNodeVector(std::shared_ptr<GraphView> graphView, bool newSchedule, bool verbose) { std::vector<std::shared_ptr<Node>> nodeVector; - - SequentialScheduler scheduler(graphView); - - if (newSchedule) - { - scheduler.resetScheduling(); - scheduler.generateScheduling(); // old way : scheduler.forward(); - } - - nodeVector = scheduler.getStaticScheduling(); + nodeVector = graphView->getOrderedNodes(); fixScheduling(nodeVector); nodeVector = removeMatchingNodes(nodeVector, "Producer"); diff --git a/src/QAT/QAT_FixedQ.cpp b/src/QAT/QAT_FixedQ.cpp index 6ada53239f92d19f96dc87e0b91247aa093caecf..2c8c44eb6ad0cb142805f9e3fbc5bd8b482694ea 100644 --- a/src/QAT/QAT_FixedQ.cpp +++ b/src/QAT/QAT_FixedQ.cpp @@ -152,9 +152,7 @@ void QuantFixedQ::insertAndInitQuantizers(std::shared_ptr<GraphView> graphView, void QuantFixedQ::devQAT(std::shared_ptr<GraphView> graphView) { - SequentialScheduler scheduler(graphView); - scheduler.generateScheduling(); - auto s = scheduler.getStaticScheduling(); + auto s = graphView->getOrderedNodes(); for (std::shared_ptr<Node> node : s) Log::info(" name : {} ", node->name()); }