每个Tensorflow GPU内核的测试在哪里？

Question

我想编写自己的操作系统，知道代码库中的测试位置会很有用。我知道tensorflow / core / kernels是内核实现的所在。此目录中的* _test.cc文件是否正在测试GPU内核实现？我查看了这些文件，似乎很多测试都在CPU上运行（我在大多数DEVICE_CPU调用中看到CreateOpKernel）。还有其他测试吗？

Answer 1

要查看tensorflow / core / kernels的特定gpu（cuda）测试，请查看tensorflow / core / kernels目录中的BUILD文件，并查找以下开头的节：

tf_cuda_cc_test

那些定义了每个cuda特定的测试。

为了更容易使用cuda支持运行所有cc测试，提供了一个示例script/bazel命令。