traversal

Author: justinchuby

olive/passes/onnx/mnb_to_qdq.py

@@ -151,7 +151,7 @@ def mat_mul_n_bits_replacement(
         # TODO(justinchuby): Call the rewriter with replace_mat_mul_n_bits
 
         # 2. Repack the quantized weights
-        for node in ir_model.graph:
+        for node in ir.traversal.RecursiveGraphIterator(ir_model.graph):
             if "needs_repacking" not in node.meta:
                 continue