Bazel cuda_add_executable相当于？

Question

我正在探索使用Bazel而不是cmake来构建项目。 cmake中有“cuda_add_executable”或“cuda_add_library”吗？

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

Answer 1

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