3

I was exploring using Bazel instead of cmake for building a project. Is there something equivalent to "cuda_add_executable" or "cuda_add_library" in cmake?

cc_library doesn't appear to allow ".cu" extensions, so that would require a rename of those files, and looking through the tensorflow repo, it's not entirely clear to me how nvcc might be enabled.

talonmies
  • 70,661
  • 34
  • 192
  • 269
dfreese
  • 378
  • 1
  • 10

1 Answers1

3

There is no such rule provided by the Bazel team. However you can build one using our extension language. Tensorflow does a bit different. It configure a crosstool and pass it .cc files.

Damien Martin-Guillerez
  • 2,260
  • 12
  • 12