From 164c9a093636945901278f243ad07c4fe9f420ac Mon Sep 17 00:00:00 2001 From: Dongju Chae Date: Thu, 20 Aug 2020 11:00:04 +0900 Subject: [PATCH] [FIX/DMABUF] Fix external segments handling This patch fixes external segments handling. It was creating only one external segment. Signed-off-by: Dongju Chae --- src/core/ne-segment-table.cc | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/core/ne-segment-table.cc b/src/core/ne-segment-table.cc index 82dd073..dcb47ff 100644 --- a/src/core/ne-segment-table.cc +++ b/src/core/ne-segment-table.cc @@ -184,8 +184,10 @@ SegmentTable::createExternalInputSegments (const Model *model, uint32_t offset = meta->getInputSegmentOffset (i); uint32_t size = meta->getSegmentSize (idx); - return createExternalSegment (idx, size, offset, + int status = createExternalSegment (idx, size, offset, input->bufs[i].dmabuf); + if (status != 0) + return status; } } @@ -208,8 +210,10 @@ SegmentTable::createExternalOutputSegments (const Model *model, uint32_t offset = meta->getOutputSegmentOffset (i); uint32_t size = meta->getSegmentSize (idx); - return createExternalSegment (idx, size, offset, + int status = createExternalSegment (idx, size, offset, output->bufs[i].dmabuf); + if (status != 0) + return status; } } -- 2.7.4