Function TEST

test/cpp/phi/api/test_scale_benchmark.cc:31–61

namespace tests {

TEST(API, scale) {
  auto x = experimental::full({3, 4}, 1.0, phi::DataType::FLOAT32, CPUPlace());

  const size_t cycles = 300;
  phi::tests::Timer timer;
  double t1{}, t2{}, t3{};

  for (size_t i = 0; i < cycles; ++i) {
    timer.tic();
    for (size_t i = 0; i < cycles; ++i) {
      auto out = experimental::scale_kernel_context(x, 2.0, 1.0, true);
    }
    t1 += timer.toc();

    timer.tic();
    for (size_t i = 0; i < cycles; ++i) {
      auto out = experimental::scale(x, 2.0, 1.0, true);
    }
    t2 += timer.toc();

    timer.tic();
    for (size_t i = 0; i < cycles; ++i) {
      auto out = experimental::scale_switch_case(x, 2.0, 1.0, true);
    }
    t3 += timer.toc();
  }

  LOG(INFO) << "The cost of kernel_context is " << t1 << "ms.";
  LOG(INFO) << "The cost of variadic_args_kernel_fn is " << t2 << "ms.";
  LOG(INFO) << "The cost of switch_case is " << t3 << "ms.";
}

}  // namespace tests
}  // namespace paddle