Question

tensorflow中已经有一些功能可以创建基准，可以在行动中看到in the adjust contrast op benchmark。但是，如果我在我的机器上运行它，我只得到一个空输出：

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

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

1 个答案: