mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix void pointer arithmetic (#98)

pull/12/head
q.yao 2021-09-27 10:52:26 +08:00 committed by GitHub
parent b8217d8243
commit 067b915383
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend_ops/tensorrt/instance_norm/trt_instance_norm.cpp
View File

@ -62,15 +62,15 @@ int TRTInstanceNormalization::enqueue(
int elem_size = getElementSize(inputDesc[1].type);
void* n_scales = (void*)workspace;
void* n_bias = (void*)(workspace + getAlignedSize(n * c * elem_size));
void* n_bias = (void*)((char*)workspace + getAlignedSize(n * c * elem_size));
const void* scales = (const void*)inputs[1];
const void* bias = (const void*)inputs[2];
for (int i = 0; i < n; ++i) {
cudaMemcpyAsync(n_scales + i * c * elem_size, scales, c * elem_size,
cudaMemcpyAsync((char*)n_scales + i * c * elem_size, scales, c * elem_size,
cudaMemcpyDeviceToDevice, stream);
cudaMemcpyAsync(n_bias + i * c * elem_size, bias, c * elem_size,
cudaMemcpyAsync((char*)n_bias + i * c * elem_size, bias, c * elem_size,
cudaMemcpyDeviceToDevice, stream);
}

2
backend_ops/tensorrt/roi_align/trt_roi_align.cpp
View File

@ -113,7 +113,7 @@ int TRTRoIAlign::enqueue(const nvinfer1::PluginTensorDesc *inputDesc,
switch (mPoolMode) {
case 0: // max
argmax_y = workSpace;
argmax_x = argmax_y + output_size * word_size;
argmax_x = (char *)argmax_y + output_size * word_size;
break;
case 1: // avg
break;