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 does a bit different. It configure a crosstool 构建一个并传递 .cc 文件。