genn-team / ml_genn

A library for deep learning with Spiking Neural Networks (SNN).
https://ml-genn.readthedocs.io
GNU Lesser General Public License v2.1
24 stars 7 forks source link

Make better use of VarLocation #121

Open neworderofjamie opened 4 days ago

neworderofjamie commented 4 days ago

While it's fair to create user-facing model variables using default VarLocation::HOST_DEVICE, stuff like EventProp ring-buffers can be quite big and, as they get created pinned, can easily take down your host if you don't have lots of free memory. Would be good if compilers could create these VarLocation::DEVICE