commit 91dfbf483abbf0520a4095d21ead4f7f5f02aa19
author Daniel Garvey <34486624+dan-garvey@users.noreply.github.com> Fri Oct 06 20:23:13 2023 -0500
committer GitHub <noreply@github.com> Sat Oct 07 01:23:13 2023 +0000
tree f17fc6c9c2aca01faac157a8d6279c0fa39d73f1
parent 5121c3ff4932b72c804404a560ba303bb9c94b22

Adds torch-mlir dequant ops to the lowering pipeline (#15128)

These should be fine even if we don't have dequant ops

compiler/plugins/input/Torch/torch-iree/InputConversion/Passes.cpp

diff --git a/compiler/plugins/input/Torch/torch-iree/InputConversion/Passes.cpp b/compiler/plugins/input/Torch/torch-iree/InputConversion/Passes.cpp
index b943851..fdafc92 100644
--- a/compiler/plugins/input/Torch/torch-iree/InputConversion/Passes.cpp
+++ b/compiler/plugins/input/Torch/torch-iree/InputConversion/Passes.cpp
@@ -42,7 +42,10 @@
     // now be simplified.
     pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
   }
-
+  pm.addNestedPass<func::FuncOp>(
+      torch::TorchConversion::createUnpackQuantTensorPass());
+  pm.addNestedPass<func::FuncOp>(
+      mlir::torch::TorchConversion::createConvertCustomQuantOpPass());
   pm.addNestedPass<func::FuncOp>(
       torch::Torch::createDecomposeComplexOpsPass(emptyArrayRef));
   pm.addNestedPass<func::FuncOp>(torch::createConvertTorchToTMTensorPass());