Merge pull request #426 from SushmitaThakallapalli1980/sush.remove_qdq_binary

remove_qdq_binary

Author: SushmitaThakallapalli1980

src/Compiler/OnnxToMlirPasses.cpp

@@ -84,6 +84,10 @@ void addONNXToMLIRPasses(mlir::PassManager &pm, bool targetCPU,
   pm.addPass(onnx_mlir::createSimplifyShapeRelatedOpsPass(
       opts.enableQuarkQuantizedLegalization));
 
+  // Pass for removing binary ops if one of the inputs is fed by a Constant
+  if (opts.enableRemoveBinary)
+    pm.addPass(createFoldDQBinaryQPass());
+
   // Pass for removing Dq and Q around data movement in Dq->op->Q Ops chain
   if (opts.enableRemoveDqQAroundOp)
     pm.addPass(createQDQAroundOpOptONNXToONNXPass());

src/Compiler/OnnxToMlirPasses.hpp

@@ -18,6 +18,7 @@ struct OnnxToMlirOptions {
   bool enableConvTranspose1dDecomposeToPhasedConv = false;
   bool enableRemoveDqQOp = true;
   bool enableRemoveDqQAroundOp = true;
+  bool enableRemoveBinary = false;
 
   bool disableRecomposeOption = false;
   bool enableONNXHybridPass = true;

src/Dialect/ONNX/Transforms/CMakeLists.txt

@@ -43,7 +43,8 @@ add_onnx_mlir_library(OMInstrumentONNX
 add_onnx_mlir_library(OMONNXRewrite
   ConstProp.cpp
   QDQAroundOpOpt.cpp    
-  QDQOpt.cpp    
+  QDQOpt.cpp
+  DQBinaryQOpt.cpp  
   ConvOpt.cpp
   Decompose.cpp
   DecomposeEinsum.cpp