Bazel cuda_add_executable 等价物？

Question

我正在探索使用 Bazel 而不是 cmake 来构建项目。 cmake中是否有相当于“cuda_add_executable”或“cuda_add_library”的东西？

cc_library 似乎不允许“.cu”扩展名，因此需要重命名这些文件，并且查看 tensorflow 存储库，我并不完全清楚如何启用 nvcc。

Answer 1

Bazel 团队没有提供这样的规则。但是，您可以使用我们的extension language 构建一个。 Tensorflow 的做法有点不同。它configure a crosstool 并传递给它 .cc 文件。