rewriter.eraseOp(lhs_mul);
    rewriter.eraseOp(rhs_mul);

    return success();
  }
};

class OutlineCompositesPass
    : public impl::OutlineCompositesPassBase<OutlineCompositesPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(OutlineCompositesPass)

  void runOnOperation() override {
    auto func = getOperation();
    RewritePatternSet patterns(&getContext());