search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.56k stars 394 forks source link

Advertising bazel-contrib/rules_cuda #1149

Open cloudhan opened 1 year ago

cloudhan commented 1 year ago

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.

tpopp commented 1 year ago

This is not yet feasible as XLA and Tensorflow are still tied together at Google, sharing the same code, but it might be possible in 1-2 months as they are more completely separated.