!5578 Fix sparse slicer leak

Merge pull request !5578 from ZPaC/master-fix-sparse-slice-leak
2020-08-31 19:55:43 +08:00 · 2020-08-31 19:55:43 +08:00 · a1b79288c3
parent 042ac51f05 b0b3cd61bf
commit a1b79288c3
1 changed files with 5 additions and 0 deletions
--- a/mindspore/ccsrc/frontend/parallel/ps/worker_proxy.h
+++ b/mindspore/ccsrc/frontend/parallel/ps/worker_proxy.h
@ -502,6 +502,11 @@ void WorkerProxy<T>::SparseSlicer(int timestamp, const ::ps::KVPairs<T> &send, c

      kvs.lens = reduced_lens;
      kvs.vals = reduced_data;
+
+      delete[] src_grad_data;
+      delete[] src_indice_data;
+      delete[] new_grad;
+      delete[] new_indices;
    }

    if (indices_size <= 0) {