Skip to content

Commit bdc7ffb

Browse files
authored
[TE] fix: deleting buffer that doesn't belong to us (#739)
* Fix deleting buffer that doesn't belong to us * Linting
1 parent 55afeb7 commit bdc7ffb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

mooncake-transfer-engine/src/transport/tcp_transport/tcp_transport.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,9 @@ struct Session : public std::enable_shared_from_this<Session> {
160160
[this, addr, dram_buffer, self](const asio::error_code &ec,
161161
std::size_t transferred_bytes) {
162162
#ifdef USE_CUDA
163-
delete[] dram_buffer;
163+
if (isCudaMemory(addr)) {
164+
delete[] dram_buffer;
165+
}
164166
#endif
165167
if (ec) {
166168
LOG(ERROR)

0 commit comments

Comments
 (0)