add recompose to pass

raayandhar · raayandhar · commit ea723c136a35 · 2025-10-14T16:47:15.000Z
Signed-off-by: raayandhar &lt;rdhar@amd.com&gt;
diff --git a/lib/Dialect/Torch/Transforms/Passes.cpp b/lib/Dialect/Torch/Transforms/Passes.cpp
@@ -76,6 +76,7 @@ void mlir::torch::Torch::createTorchDynamoExportToTorchBackendPipeline(
   if (options.decompose) {
     pm.addNestedPass<func::FuncOp>(
         Torch::createDecomposeComplexOpsPass(options.backendLegalOps));
+    pm.addNestedPass<func::FuncOp>(Torch::createRecomposeComplexOpsPass());
     pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,7 @@ void mlir::torch::Torch::createTorchDynamoExportToTorchBackendPipeline(`
`76`	`76`	`if (options.decompose) {`
`77`	`77`	`pm.addNestedPass<func::FuncOp>(`
`78`	`78`	`Torch::createDecomposeComplexOpsPass(options.backendLegalOps));`
	`79`	`+ pm.addNestedPass<func::FuncOp>(Torch::createRecomposeComplexOpsPass());`
`79`	`80`	`pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());`
`80`	`81`	`}`
`81`	`82`	`}`