Merge branch 'cleanup-give-unique-parameter-tensors' into dev

iksnagreb · Jan 28, 2025 · 2a59cae · 2a59cae
2 parents 5bbbda7 + 109bc2b
commit 2a59cae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/qonnx/core/modelwrapper.py b/src/qonnx/core/modelwrapper.py
@@ -41,6 +41,7 @@
 from qonnx.transformation.general import (
     RemoveStaticGraphInputs,
     RemoveUnusedTensors,
+    GiveUniqueParameterTensors,
     SortCommutativeInputsInitializerLast,
     SortGraph,
 )
@@ -155,6 +156,7 @@ def cleanup(self):
             RemoveStaticGraphInputs(),
             SortGraph(),
             SortCommutativeInputsInitializerLast(),
+            GiveUniqueParameterTensors(),
         ]
         for trn in cleanup_transforms:
             transformed_model = transformed_model.transform(trn, cleanup=False, make_deepcopy=False)