Skip to content

Commit 932bcd6

Browse files
committed
Avoid arithmetic on void pointer in multilevelProposeROIPlugin.cpp #1028
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
1 parent 1fe5532 commit 932bcd6

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

plugin/multilevelProposeROI/multilevelProposeROIPlugin.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -406,7 +406,7 @@ int MultilevelProposeROI::enqueue(
406406
static_cast<float>(mImageSize.d[1]), // Input Height
407407
static_cast<float>(mImageSize.d[2]),
408408
DataType::kFLOAT, // mType,
409-
mParam, proposal_ws, workspace + kernel_workspace_offset,
409+
mParam, proposal_ws, static_cast<uint8_t*>(workspace) + kernel_workspace_offset,
410410
inputs[2 * i + 1], // inputs[object_score],
411411
inputs[2 * i], // inputs[bbox_delta]
412412
mValidCnt->mPtr,
@@ -419,7 +419,7 @@ int MultilevelProposeROI::enqueue(
419419

420420
ConcatTopKWorkSpace ctopk_ws(batch_size, mFeatureCnt, mKeepTopK, mType);
421421
status = ConcatTopK(stream, batch_size, mFeatureCnt, mKeepTopK, DataType::kFLOAT,
422-
workspace + kernel_workspace_offset, ctopk_ws, reinterpret_cast<void**>(mDeviceScores),
422+
static_cast<uint8_t*>(workspace) + kernel_workspace_offset, ctopk_ws, reinterpret_cast<void**>(mDeviceScores),
423423
reinterpret_cast<void**>(mDeviceBboxes), final_proposals);
424424

425425
assert(status == cudaSuccess);

0 commit comments

Comments
 (0)