xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon openxla

Advertising bazel-contrib/rules_cuda

Open cloudhan opened this issue 2 years ago • 0 comments

rules_cuda is a community effort for adding cuda support for bazel.

It currently supports:

  1. linux and windows
  2. nvcc+gcc or nvcc+msvc where cuda >= 10.0 or standalone clang cuda
  3. relocatable device code (rdc) and device link time optimiaztion (dlto)

It is also purely implemented in starlark. Toolchains can be configured with cc_toolchain style DSL. Attacking TF repo might be too ambitious at the moment, I think OpenXLA might suit for it.

cloudhan avatar Feb 07 '23 08:02 cloudhan