Browse Source

Fix sparse-slicer leak.

tags/v1.0.0
ZPaC 5 years ago
parent
commit
b0b3cd61bf
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      mindspore/ccsrc/frontend/parallel/ps/worker_proxy.h

+ 5
- 0
mindspore/ccsrc/frontend/parallel/ps/worker_proxy.h View File

@@ -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) {


Loading…
Cancel
Save