Bazel cuda_add_executable équivalent?

Question

J'explorais l'utilisation de Bazel au lieu de cmake pour construire un projet. Y a-t-il quelque chose d'équivalent à "cuda_add_executable" ou "cuda_add_library" dans cmake? Cc_library ne semble pas autoriser les extensions ".cu", ce qui nécessiterait un changement de nom de ces fichiers, et en regardant dans le repo tensorflow, il n'est pas tout à fait clair pour moi comment nvcc pourrait être activé.

Answer 1

Il n'existe aucune règle de ce type fournie par l'équipe Bazel. Cependant, vous pouvez en construire un en utilisant notre extension language. Tensorflow fait un peu différent. Il configure a crosstool et lui passer les fichiers .cc.