Limit unrolling

EnzymeAD · Mar 10, 2024 · e98d0a1 · e98d0a1
1 parent f170798
commit e98d0a1
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/enzyme_ad/jax/Passes/EnzymeHLOUnroll.cpp b/src/enzyme_ad/jax/Passes/EnzymeHLOUnroll.cpp
@@ -128,9 +128,6 @@ struct EnzymeHLOUnrollPass
     auto context = getOperation()->getContext();
     RewritePatternSet patterns(context);
     patterns.add<WhileUnroll>(context);
-    mlir::stablehlo::populateStablehloCanonicalizationPatterns(context,
-                                                               &patterns);
-
     GreedyRewriteConfig config;
     if (failed(applyPatternsAndFoldGreedily(getOperation(), std::move(patterns),
                                             config))) {