Fix PbTensor memory handle calculation

Fix shm handle on loading PbTensor pb_memory
triton-inference-server · May 21, 2024 · be163e6 · be163e6
1 parent b3afcc9
commit be163e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(