search

elixir-nx / xla

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

Add Docker build script for Linux ARM #60

Closed jonatanklosko closed 8 months ago

jonatanklosko commented 8 months ago

Closes #59.

The CI cross-compilation for ARM no longer works as expected, so I added Docker scripts instead. I built it in the cloud both for CPU and CUDA and added all of them to the latest release.