From be163e6fe4cb8320649705cabc70259417d2abfb Mon Sep 17 00:00:00 2001 From: Pawel Ziecina Date: Tue, 21 May 2024 09:21:38 +0200 Subject: [PATCH] Fix PbTensor memory handle calculation Fix shm handle on loading PbTensor pb_memory --- src/pb_tensor.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pb_tensor.cc b/src/pb_tensor.cc index d9d47784..67afff29 100644 --- a/src/pb_tensor.cc +++ b/src/pb_tensor.cc @@ -585,7 +585,7 @@ PbTensor::LoadFromSharedMemory( if (tensor_shm_ptr->memory == 0) { std::size_t pb_memory_offset = name_offset + name_shm->Size(); pb_memory = PbMemory::LoadFromSharedMemory( - shm_pool, pb_memory_offset, tensor_shm.data_.get() + pb_memory_offset, + shm_pool, tensor_handle + pb_memory_offset, tensor_shm.data_.get() + pb_memory_offset, open_cuda_handle); } else { pb_memory = PbMemory::LoadFromSharedMemory(