[luci] Copy qparam to circle output (Samsung#12855)

This supports copying qparam to circle output. ONE-DCO-1.0-Signed-off-by: Hyukjin Jeong <[email protected]>
lemmaa · Apr 9, 2024 · 57351d0 · 57351d0
1 parent 6de58be
commit 57351d0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/compiler/luci/pass/src/CopyQuantParamPass.cpp b/compiler/luci/pass/src/CopyQuantParamPass.cpp
@@ -71,6 +71,12 @@ bool CopyQuantParamPass::run(loco::Graph *g)
 
     copy_quantparam(nodes.src, nodes.dst);
 
+    if (auto output = dynamic_cast<luci::CircleOutput *>(nodes.dst))
+    {
+      auto from_node = loco::must_cast<luci::CircleNode *>(output->from());
+      copy_quantparam(output, from_node);
+    }
+
     INFO(l) << "Quantparam of " << src << " is copied to " << dst << std::endl;
   }