diff --git a/compiler/plugins/input/Torch/InputConversion/Passes.cpp b/compiler/plugins/input/Torch/InputConversion/Passes.cpp index 8f51c617a357..00ab1a444854 100644 --- a/compiler/plugins/input/Torch/InputConversion/Passes.cpp +++ b/compiler/plugins/input/Torch/InputConversion/Passes.cpp @@ -50,6 +50,7 @@ void createTorchToIREEPipeline( pm.addNestedPass( torch::Torch::createDecomposeComplexOpsPass(emptyArrayRef)); pm.addNestedPass(torch::Torch::createFuseQuantizedOpsPass()); + pm.addNestedPass(createCanonicalizerPass()); pm.addNestedPass(torch::Torch::createScalarizeShapesPass()); pm.addNestedPass(torch::createConvertTorchToTMTensorPass()); pm.addNestedPass(