forked from triton-inference-server/python_backend
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpb_string.cc
More file actions
126 lines (108 loc) · 4.41 KB
/
pb_string.cc
File metadata and controls
126 lines (108 loc) · 4.41 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// Copyright 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
// * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
// * Neither the name of NVIDIA CORPORATION nor the names of its
// contributors may be used to endorse or promote products derived
// from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
// OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#include "pb_string.h"
namespace triton { namespace backend { namespace python {
std::unique_ptr<PbString>
PbString::Create(
std::unique_ptr<SharedMemoryManager>& shm_pool, const std::string& string)
{
AllocatedSharedMemory<StringShm> string_container_shm =
shm_pool->Construct<StringShm>();
string_container_shm.data_->length = string.size();
AllocatedSharedMemory<char> string_shm =
shm_pool->Construct<char>(string.size());
std::memcpy(string_shm.data_.get(), string.data(), string.size());
string_container_shm.data_->data = string_shm.handle_;
return std::unique_ptr<PbString>(
new PbString(string_container_shm, string_shm));
}
std::unique_ptr<PbString>
PbString::Create(
const std::string& string, char* data_shm,
bi::managed_external_buffer::handle_t handle)
{
StringShm* string_container_shm = reinterpret_cast<StringShm*>(data_shm);
string_container_shm->length = string.size();
char* string_shm = data_shm + sizeof(StringShm);
std::memcpy(string_shm, string.data(), string.size());
return std::unique_ptr<PbString>(
new PbString(string_container_shm, string_shm, handle));
}
std::unique_ptr<PbString>
PbString::LoadFromSharedMemory(
std::unique_ptr<SharedMemoryManager>& shm_pool,
bi::managed_external_buffer::handle_t handle)
{
AllocatedSharedMemory<StringShm> string_container_shm =
shm_pool->Load<StringShm>(handle);
AllocatedSharedMemory<char> string_shm =
shm_pool->Load<char>(string_container_shm.data_->data);
return std::unique_ptr<PbString>(
new PbString(string_container_shm, string_shm));
}
std::unique_ptr<PbString>
PbString::LoadFromSharedMemory(
bi::managed_external_buffer::handle_t handle, char* data_shm)
{
StringShm* string_container_shm = reinterpret_cast<StringShm*>(data_shm);
char* string_shm = data_shm + sizeof(StringShm);
return std::unique_ptr<PbString>(
new PbString(string_container_shm, string_shm, handle));
}
PbString::PbString(
AllocatedSharedMemory<StringShm>& string_container_shm,
AllocatedSharedMemory<char>& string_shm)
: string_container_shm_(std::move(string_container_shm)),
string_shm_(std::move(string_shm))
{
string_shm_ptr_ = string_shm_.data_.get();
string_container_shm_ptr_ = string_container_shm_.data_.get();
string_handle_ = string_container_shm_.handle_;
}
PbString::PbString(
StringShm* string_container_shm, char* string_shm,
bi::managed_external_buffer::handle_t handle)
{
string_shm_ptr_ = string_shm;
string_container_shm_ptr_ = string_container_shm;
string_handle_ = handle;
}
bi::managed_external_buffer::handle_t
PbString::ShmHandle()
{
return string_handle_;
}
std::size_t
PbString::ShmStructSize(const std::string& string)
{
return string.size() + sizeof(StringShm);
}
std::size_t
PbString::Size()
{
return string_container_shm_ptr_->length + sizeof(StringShm);
}
}}} // namespace triton::backend::python