search

elixir-nx / bumblebee

Pre-trained Neural Network models in Axon (+ 🤗 Models integration)
Apache License 2.0
1.33k stars 96 forks source link

Speed up tests #230

Closed jonatanklosko closed 1 year ago

jonatanklosko commented 1 year ago

Mostly removing EXLA-compilation from places where EXLA backend is faster for one-off runs.

jonatanklosko commented 1 year ago

This combination of tests fails to run on CI. I thought it may be because with backend it's more involved (no fusion), but I checked and it fails without the changes or even when using EXLA globally. From the logs it's not even obvious what limit is the issue, so I will merge as this is at least faster locally.