StringStream stream;
    op->print(stream, mlir::OpPrintingFlags().useLocalScope());
    ops_seen_by_pass_[pass] = stream.ss.str();
  }
  void runAfterPass(Pass* pass, Operation* op) override {
    StringStream stream;
    op->print(stream, mlir::OpPrintingFlags().useLocalScope());
    if (!absl::StrContains(stream.ss.str(), kTestInstrumentationSearch) &&

  std::string mlir_output;
  llvm::raw_string_ostream output_stream(mlir_output);
  // Dump MLIR with eliding large elements.
  module->print(
      output_stream,
      mlir::OpPrintingFlags().useLocalScope().elideLargeElementsAttrs());
  return mlir_output;
}

  std::string expected_txt_module;
  {
    llvm::raw_string_ostream os(expected_txt_module);
    module_ref->getOperation()->print(os,
                                      mlir::OpPrintingFlags().useLocalScope());
    os.flush();
  }

  std::string filepath = DumpMlirOpToFile("module", module_ref.get());
  ASSERT_NE(filepath, "(TF_DUMP_GRAPH_PREFIX not specified)");
  ASSERT_NE(filepath, "LOG(INFO)");

  }
};

}  // namespace

void EnableIrPrinting(mlir::PassManager& pm,
                      absl::string_view file_name_prefix) {
  mlir::OpPrintingFlags flag{};
  flag.useLocalScope().elideLargeElementsAttrs().enableDebugInfo();

  // IR printing requires multithreading disabled.
  // Even if multithreading is already disabled, if we are executing within a

  LOG(INFO) << "Dumping MLIR operation '" << op->getName().getStringRef().str()
            << "' to '" << filepath << "'";
  if (pass_manager) PrintPassPipeline(*pass_manager, op, *os);
  op->print(*os, mlir::OpPrintingFlags().useLocalScope());
  return filepath;
}

std::string GetDumpDirFromEnvVar() {
  const char* prefix_env = getenv("TF_DUMP_GRAPH_PREFIX");
  if (!prefix_env) {
    LOG(WARNING)