Add support for overriding the target platform

elixir-nx / xla

Pre-compiled XLA extension

Apache License 2.0

83 stars 21 forks source link

Add support for overriding the target platform #38

Closed jonatanklosko closed 1 year ago

jonatanklosko commented 1 year ago

When cross-compiling EXLA with Nerves we want to override the XLA archive architecture.

$ mix xla.info archive_filename
xla_extension-aarch64-darwin-cpu.tar.gz
$ XLA_TARGET_PLATFORM=aarch64-linux-gcc mix xla.info archive_filename
xla_extension-aarch64-linux-gcc-cpu.tar.gz