search

ahrefs / ocannl

OCANNL: OCaml Compiles Algorithms for Neural Networks Learning
BSD 2-Clause "Simplified" License
67 stars 2 forks source link

Make `unsafe_cleanup` safe, get rid of `unsafe_shutdown`, initialize devices on-demand #274

Closed lukstafi closed 3 months ago

lukstafi commented 3 months ago

There's no benefit to the shutdown flag, it's better to build new physical device objects on demand. (Always shutdown, but don't make a fuss about it.)

lukstafi commented 3 months ago

We cannot make unsafe_cleanup safe, since typically there'll still be live contexts when it's called.