| 85 | } |
| 86 | |
| 87 | void BM_cpu_RandomGamma(int iters, int nsamp, int nalpha) { |
| 88 | testing::ItemsProcessed(static_cast<int64>(iters) * nsamp * nalpha); |
| 89 | Graph* g = new Graph(OpRegistry::Global()); |
| 90 | test::graph::RandomGamma(g, test::graph::Constant(g, VecShape(nsamp)), |
| 91 | test::graph::Constant(g, VecAlphas(nalpha))); |
| 92 | test::Benchmark("cpu", g).Run(iters); |
| 93 | } |
| 94 | BENCHMARK(BM_cpu_RandomGamma)->RangePair(1 << 14, 4 << 15, 2, 50); |
| 95 | |
| 96 | void BM_PhiloxRandom(int iters) { |
nothing calls this directly
no test coverage detected