Support half precision in the GCCJIT backend

ahrefs / ocannl

OCANNL: OCaml Compiles Algorithms for Neural Networks Learning

BSD 2-Clause "Simplified" License

69 stars 2 forks source link

Support half precision in the GCCJIT backend #284

Open lukstafi opened 2 months ago

lukstafi commented 2 months ago

The big blocker is support for half-precision and precision conversions in ocaml-gccjit. Plus a smaller issue to have API support for taking a bigarray pointer without using Ctypes.bigarray_start, because ctypes does not support half precision.

This is a companion issue to https://github.com/lukstafi/ocaml-gccjit/issues/11