!5421 Fix sparse slicer leak

Merge pull request !5421 from ZPaC/r0.7-fix-sparseslicer-leak
2020-08-28 16:09:32 +08:00 · 2020-08-28 16:09:32 +08:00 · 2ff8de674c
parent 13978a2c98 1ac075919a
commit 2ff8de674c
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) {