@@ -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