Skip to content

Commit bd14e27

Browse files
committed
Fix PbTensor memory handle calculation
Fix shm handle on loading PbTensor pb_memory
1 parent f01f891 commit bd14e27

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pb_tensor.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -585,7 +585,7 @@ PbTensor::LoadFromSharedMemory(
585585
if (tensor_shm_ptr->memory == 0) {
586586
std::size_t pb_memory_offset = name_offset + name_shm->Size();
587587
pb_memory = PbMemory::LoadFromSharedMemory(
588-
shm_pool, pb_memory_offset, tensor_shm.data_.get() + pb_memory_offset,
588+
shm_pool, tensor_handle + pb_memory_offset, tensor_shm.data_.get() + pb_memory_offset,
589589
open_cuda_handle);
590590
} else {
591591
pb_memory = PbMemory::LoadFromSharedMemory(

0 commit comments

Comments
 (0)