Function BM_StressTest

tensorflow/compiler/jit/graphcycles/graphcycles_test.cc:511–528

}

static void BM_StressTest(int iters, int num_nodes) {
  while (iters > 0) {
    tensorflow::GraphCycles g;
    int32 *nodes = new int32[num_nodes];
    for (int i = 0; i < num_nodes; i++) {
      nodes[i] = g.NewNode();
    }
    for (int i = 0; i < num_nodes && iters > 0; i++, iters--) {
      int end = std::min(num_nodes, i + 5);
      for (int j = i + 1; j < end; j++) {
        if (nodes[i] >= 0 && nodes[j] >= 0) {
          CHECK(g.InsertEdge(nodes[i], nodes[j]));
        }
      }
    }
    delete[] nodes;
  }
}
BENCHMARK(BM_StressTest)->Range(2048, 1048576);