body_arg.use_begin()->getOwner() == body_ret) {
      can_eliminate.set(result_idx);
    }
  }

  if (can_eliminate.empty()) return;

  func::FuncOp cloned_cond = CloneFunctionIfNeeded(cond);
  func::FuncOp cloned_body = CloneFunctionIfNeeded(body);
  op.setCondAttr(
      FlatSymbolRefAttr::get(op.getContext(), cloned_cond.getName()));
  op.setBodyAttr(
      FlatSymbolRefAttr::get(op.getContext(), cloned_body.getName()));