Function TEST

tensorflow/compiler/jit/mark_for_cuda_graph_mode_pass_test.cc:90–111

}

TEST(CudaGraphModeCompilationTest, Chains) {
  std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
  {
    GraphDefBuilder builder(GraphDefBuilder::kFailImmediately);
    Node* a =
        ops::SourceOp("UncompilableNullary", builder.opts().WithName("A"));
    Node* b = ops::UnaryOp("Elu", a, builder.opts().WithName("B"));
    Node* c = ops::UnaryOp("Elu", b, builder.opts().WithName("C"));
    Node* d =
        ops::UnaryOp("UncompilableUnary", c, builder.opts().WithName("D"));
    Node* e = ops::UnaryOp("Elu", d, builder.opts().WithName("E"));
    ops::UnaryOp("Elu", e, builder.opts().WithName("F"));
    TF_EXPECT_OK(GraphDefBuilderToGraph(builder, graph.get()));
  }

  TF_ASSERT_OK(MarkForCudaGraphModePassTestHelper::MarkForCompilation(&graph));
  auto clusters = GetClusters(*graph);
  EXPECT_EQ(6, clusters.size());
  EXPECT_EQ(clusters["B"], clusters["C"]);
  EXPECT_EQ(clusters["E"], clusters["F"]);
  EXPECT_EQ(clusters["B"], clusters["E"]);
}
}  // namespace
}  // namespace tensorflow