diff --git a/compiler/plugins/input/Torch/InputConversion/Passes.cpp b/compiler/plugins/input/Torch/InputConversion/Passes.cpp index 2dc4a9291d0c6..293921892b26d 100644 --- a/compiler/plugins/input/Torch/InputConversion/Passes.cpp +++ b/compiler/plugins/input/Torch/InputConversion/Passes.cpp @@ -49,6 +49,8 @@ void createTorchToIREEPipeline( mlir::torch::TorchConversion::createConvertCustomQuantOpPass()); pm.addNestedPass( torch::Torch::createDecomposeComplexOpsPass(emptyArrayRef)); + pm.addNestedPass(torch::Torch::createFuseQuantizedOpsPass()); + pm.addNestedPass(torch::Torch::createScalarizeShapesPass()); pm.addNestedPass(torch::createConvertTorchToTMTensorPass()); pm.addNestedPass( TorchInput::createConvertTMTensorToLinalgExtPass());