search

elixir-nx / xla

Pre-compiled XLA extension
Apache License 2.0
87 stars 20 forks source link

Cross-compile the binary for linux arm64 on CI #37

Closed jonatanklosko closed 1 year ago

jonatanklosko commented 1 year ago

Closes #36.

It's not as simple for CUDA though, I think we would need arm CUDA libs in the x86 Docker container.