Skip to content

Commit 6f32e50

Browse files
thmcaThomas McAdams
andauthored
Adding the fixes for arm32 (#120)
Co-authored-by: Thomas McAdams <thomas.mcadams@cysca.ca>
1 parent 7735f3a commit 6f32e50

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/backend_model.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1322,7 +1322,7 @@ TRITONBACKEND_InputBuffer(
13221322
InferenceRequest::Input* ti =
13231323
reinterpret_cast<InferenceRequest::Input*>(input);
13241324
Status status = ti->DataBuffer(
1325-
index, buffer, buffer_byte_size, memory_type, memory_type_id);
1325+
index, buffer, reinterpret_cast<size_t*>(buffer_byte_size), memory_type, memory_type_id);
13261326
if (!status.IsOk()) {
13271327
*buffer = nullptr;
13281328
*buffer_byte_size = 0;
@@ -1362,9 +1362,9 @@ TRITONBACKEND_InputBufferForHostPolicy(
13621362
Status status =
13631363
(host_policy_name == nullptr)
13641364
? ti->DataBuffer(
1365-
index, buffer, buffer_byte_size, memory_type, memory_type_id)
1365+
index, buffer, reinterpret_cast<size_t*>(buffer_byte_size), memory_type, memory_type_id)
13661366
: ti->DataBufferForHostPolicy(
1367-
index, buffer, buffer_byte_size, memory_type, memory_type_id,
1367+
index, buffer, reinterpret_cast<size_t*>(buffer_byte_size), memory_type, memory_type_id,
13681368
host_policy_name);
13691369
if (!status.IsOk()) {
13701370
*buffer = nullptr;

0 commit comments

Comments
 (0)