elixir-nx / ortex

ONNX Runtime bindings for Elixir
MIT License
122 stars 15 forks source link

add arm64 target #13

Closed brycedev closed 1 year ago

brycedev commented 1 year ago

i've specified an additional rpath flag for linux arm64 targets. ortex will now compile on my tested device: Arm Cortex A76

mortont commented 1 year ago

Thank you! Did you also confirm it downloads the correct libonnxruntime.so as well? Or are you providing your own?

brycedev commented 1 year ago

@mortont let me confirm this, as i did forget to unlink my own!

gBillal commented 1 year ago

I can confirm that this works on my Jetson Nano Orin.

mortont commented 1 year ago

Great, thanks for testing. I assume it's just using the CPU EP on your Jetson for now, but CUDA support on arm64 is something I'd like to add soon. Should be just a matter of a custom build of libonnxruntime targeting that combo.

brycedev commented 1 year ago

thank you so much @gBillal