如何在tensorflow中为自定义内核创建/运行基准测试

Question

tensorflow中已经有一些创建基准的功能，这些功能可以在实际操作中看到，例如在adjust contrast op基准中。然而，如果我在我的机器上运行这个，我只会得到一个空输出：

panmari@dingle:~/tensorflow$ bazel run //tensorflow/core:kernels_adjust_contrast_op_benchmark_test --test_output=all --cache_test_results=no -- --benchmarks=1000
INFO: Found 1 target...
Target //tensorflow/core:kernels_adjust_contrast_op_benchmark_test up-to-date:
  bazel-bin/tensorflow/core/kernels_adjust_contrast_op_benchmark_test
INFO: Elapsed time: 10.736s, Critical Path: 8.71s.
INFO: Running command line: bazel-bin/tensorflow/core/kernels_adjust_contrast_op_benchmark_test '--benchmarks=1000'.
Running main() from test_main.cc
Benchmark    Time(ns) Iterations
--------------------------------

我的召唤错了吗？

Answer 1

要调用基准测试，请运行以下命令（传递--benchmarks=all作为最终参数）：

$ bazel run -c opt //tensorflow/core:kernels_adjust_contrast_op_benchmark_test 
      --test_output=all --cache_test_results=no -- --benchmarks=all

要运行GPU基准测试，必须将--config=cuda传递给bazel，并将_gpu附加到测试目标的名称。例如：

$ bazel run -c opt --config=cuda  
     //tensorflow/core:kernels_adjust_contrast_op_benchmark_test_gpu 
     --test_output=all --cache_test_results=no -- --benchmarks=all