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

add single precision (float32) mode #70

Closed zjzjwang closed 6 months ago

zjzjwang commented 6 months ago

Closes #69

mostly copy from this branch vecterize

zjzjwang commented 6 months ago

Thank you for your insightful review, and I have made the changes accordingly. If you have any other suggestions, please feel free to let me know.

siboehm commented 6 months ago

LGTM, thanks for the PR @zjzjwang! Will try to do a release on the weekend