mlir::TF::kMlirPh1BridgeCounterV1,
                                     mlir::TF::kMlirPh1BridgeCounterTpu,
                                     "fallback_disabled", "success"),
            1);
}

TEST_F(SessionClusterTensorflowDialectTest, FailsWithMultipleSubmodules) {
  CellReader<int64_t> compilation_status(kCompilationStreamz);

                                     mlir::TF::kMlirPh1BridgeCounterV2,
                                     mlir::TF::kMlirPh1BridgeCounterTpu,
                                     "fallback_disabled", "success"),
            1);
}

TEST_F(FunctionClusterTensorflowDialectTest,
       RunsOutsideCompilationReplicatedBridge) {
  CellReader<int64_t> compilation_status(kCompilationStreamz);

  EXPECT_EQ(
      compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated,
                               mlir::TF::kMlirPh1BridgeCounterV2, "XLA_TPU_JIT",
                               "fallback_disabled", "failure"),
      1);
}

TEST_F(LowerClusterToRuntimeOpsTest, DumpsPipelinePasses) {
  std::vector<std::string> files;
  TF_ASSERT_OK(env_->GetChildren(test_dir_, &files));