Fix sparse-slicer leak.

5 years ago · b0b3cd61bf
parent 4dddee575d
commit b0b3cd61bf
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) {