diff --git a/src/shared_memory_manager.cc b/src/shared_memory_manager.cc index bc5245cab7..1064982669 100644 --- a/src/shared_memory_manager.cc +++ b/src/shared_memory_manager.cc @@ -67,8 +67,9 @@ SharedMemoryManager::GetCUDAHandle( TRITONSERVER_Error* SharedMemoryManager::GetMemoryInfo( - const std::string& name, size_t offset, void** shm_mapped_addr, - TRITONSERVER_MemoryType* memory_type, int64_t* device_id) + const std::string& name, size_t offset, size_t byte_size, + void** shm_mapped_addr, TRITONSERVER_MemoryType* memory_type, + int64_t* device_id) { return TRITONSERVER_ErrorNew( TRITONSERVER_ERROR_UNSUPPORTED,