Skip to content

Fix poor implementations + memory leaks

Olivier BICHLER requested to merge fix_reduce into dev

All CuDNN initialization and workspace memory allocation should be done only once! This MR should improve performances and solve (at least some) memory leak issues (workspace was allocated at each forward pass but never freed!).

Fixed issues:

  • ReduceMean
  • ReduceSum
  • Sub
  • Add
  • Mul
Edited by Olivier BICHLER

Merge request reports

Loading