elixir-nx / xla

Pre-compiled XLA extension
Apache License 2.0
83 stars 21 forks source link

Update supported CUDA targets #84

Closed jonatanklosko closed 1 month ago

jonatanklosko commented 1 month ago

While creating a new build for 11.8, I run into errors (same as https://github.com/tensorflow/tensorflow/issues/63356). Given that both latest Tensorflow and Jax require CUDA 12, it's fair for us to require it as well.

I also took this as an opportunity to rename "cuda120" to "cuda12" (closes #55).