Commit 8ca3ce56 authored by Yuxin Wu's avatar Yuxin Wu

[ZMQ] fix memory leak (#362)

parent 8e8e0b3b
......@@ -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();
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment