Incorporate cross-stream-sharing into the memory model in `Tnode`

ahrefs / ocannl

OCANNL: OCaml Compiles Algorithms for Neural Networks Learning

BSD 2-Clause "Simplified" License

69 stars 2 forks source link

Incorporate cross-stream-sharing into the memory model in `Tnode` #286

Closed lukstafi closed 1 month ago

lukstafi commented 2 months ago

In particular, this will make it explicitly controllable and more visible. Currently, the classification is implicitly implemented in the CUDA backend.