signalPassFailure();
    return;
  }
  TF::OpToParallelIdsMap op_to_parallel_ids_map;
  if (failed(FillOpToParallelIdsMap(module_op, op_to_parallel_ids_map))) {
    signalPassFailure();
    return;
  }
  TF::SideEffectAnalysis side_effect_analysis(module_op,
                                              op_to_parallel_ids_map);
  for (auto func : module_op.getOps<func::FuncOp>()) {