add stablehlo-legalize-to-qdq to the pipeline

openxla · Jun 18, 2024 · 5eea628 · 5eea628
1 parent ec0a81d
commit 5eea628
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/stablehlo/reference/Api.cpp b/stablehlo/reference/Api.cpp
@@ -68,7 +68,7 @@ FailureOr<func::FuncOp> getMainFunction(ModuleOp module, StringRef mainName) {
 class DefaultInterpreterFallback : public InterpreterFallback {
  public:
   DefaultInterpreterFallback(const InterpreterConfiguration &config)
-      : config(config){};
+      : config(config) {};
 
   virtual llvm::Error operator()(Operation &op, Scope &scope,
                                  Process *process) final {

diff --git a/stablehlo/transforms/PassPipelines.cpp b/stablehlo/transforms/PassPipelines.cpp
@@ -38,6 +38,8 @@ void createStablehloRemoveDynamismPipeline(OpPassManager &pm,
 }
 
 void createStablehloLowerQuantPipeline(OpPassManager &pm) {
+  pm.addNestedPass<mlir::func::FuncOp>(
+      stablehlo::createStablehloLegalizeQuantizedOpToQDQPass());
   pm.addNestedPass<mlir::func::FuncOp>(
       stablehlo::createStablehloLegalizeQuantToIntPass());
   pm.addNestedPass<mlir::func::FuncOp>(