From 774c5f9d47f6e4c99a897ad376a628c93a4fba69 Mon Sep 17 00:00:00 2001
From: Noam ZERAH <noam.zerah@cea.fr>
Date: Wed, 5 Mar 2025 14:01:23 +0000
Subject: [PATCH] Fixing deprecated ways of getting all the nodes in graphview
---
src/PTQ/PTQ.cpp | 11 +----------
src/QAT/QAT_FixedQ.cpp | 4 +---
2 files changed, 2 insertions(+), 13 deletions(-)
diff --git a/src/PTQ/PTQ.cpp b/src/PTQ/PTQ.cpp
index 156a45d..c07816d 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 6ada532..2c8c44e 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());
}
--
GitLab