search

elixir-nx / xla

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

Fix ARM CUDA build #85

Closed jonatanklosko closed 1 month ago

jonatanklosko commented 1 month ago

Adds patches from https://github.com/tensorflow/tensorflow/issues/62490#issuecomment-2077646521 to the build process.