siboehm / lleaves

Compiler for LightGBM gradient-boosted trees, based on LLVM. Speeds up prediction by ≥10x.
https://lleaves.readthedocs.io/en/latest/
MIT License
333 stars 28 forks source link

Query on halting single precision support #69

Closed zjzjwang closed 6 months ago

zjzjwang commented 6 months ago

Hello,

I've been interested in using the single precision mode in lleaves.

I noticed there was an initial implementation in the vectorize branch (link), but development seems to have halted.

Could you share the reasons behind discontinuing this feature?

Thanks for your great work on lleaves.

siboehm commented 6 months ago

No good reason really, I just never had a usecase for it. Double precision is useful because it leads to bitequal results compared to LightGBM (in most cases). I'm happy to take a PR!