[ZMQ] fix memory leak (#362)

8ca3ce56 · Yuxin Wu · 8e8e0b3b · 8ca3ce56
Commit 8ca3ce56 authored Dec 18, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

tensorpack/user_ops/zmq_recv_op.cc tensorpack/user_ops/zmq_recv_op.cc +0 -1

No files found.
--- a/tensorpack/user_ops/zmq_recv_op.cc
+++ b/tensorpack/user_ops/zmq_recv_op.cc
@@ -74,7 +74,6 @@ class ZMQRecvOp: public AsyncOpKernel {
      auto ptr = output->bit_casted_shaped<char, 1>(
          {shape.num_elements() * DataTypeSize(recv_dtype)}).data();
      memcpy(ptr, tensors[i].buf, tensors[i].buf_size);
-      ctx->set_output(i, *output);
    }
    done();
  }