NASP/cuda-samples
35
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2024-11-24 18:59:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #182 from Wenlong-Zhu/master

Browse Source 
Fix cudaExtent.width set error.
This commit is contained in:
Rob Nertney 2023-03-27 20:53:45 -07:00 committed by GitHub
commit e612904184
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Samples/4_CUDA_Libraries/cudaNvSciNvMedia/cuda_consumer.cu
View File

@ -158,6 +158,12 @@ static void cudaImportNvSciImage(cudaExternalResInterop &cudaExtResObj,
pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_Layout; pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_Layout;
pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_PlaneBitsPerPixel; pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_PlaneBitsPerPixel;
pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_PlaneOffset; pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_PlaneOffset;
pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_PlanePitch;
pairArrayOut[numAttrs++].key = NvSciBufImageAttrKey_PlaneAlignedHeight;
uint32_t planePitchs[10];
uint32_t planePixel[10];
uint32_t planeAlignedHeight[10];
checkNvSciErrors(NvSciBufAttrListGetAttrs(attrlist, pairArrayOut, numAttrs)); checkNvSciErrors(NvSciBufAttrListGetAttrs(attrlist, pairArrayOut, numAttrs));
@ -177,6 +183,13 @@ static void cudaImportNvSciImage(cudaExternalResInterop &cudaExtResObj,
cudaExtResObj.planeCount * sizeof(int32_t)); cudaExtResObj.planeCount * sizeof(int32_t));
memcpy(cudaExtResObj.planeOffset, (uint64_t *)pairArrayOut[7].value, memcpy(cudaExtResObj.planeOffset, (uint64_t *)pairArrayOut[7].value,
cudaExtResObj.planeCount * sizeof(uint64_t)); cudaExtResObj.planeCount * sizeof(uint64_t));
memcpy(planePixel, (uint32_t *)pairArrayOut[6].value,
cudaExtResObj.planeCount * sizeof(uint32_t));
memcpy(planePitchs, (uint32_t *)pairArrayOut[8].value,
cudaExtResObj.planeCount * sizeof(uint32_t));
memcpy(planeAlignedHeight, (uint32_t *)pairArrayOut[9].value,
cudaExtResObj.planeCount * sizeof(uint32_t));
NvSciBufAttrValImageLayoutType layout = NvSciBufAttrValImageLayoutType layout =
*(NvSciBufAttrValImageLayoutType *)pairArrayOut[5].value; *(NvSciBufAttrValImageLayoutType *)pairArrayOut[5].value;
@ -201,8 +214,8 @@ static void cudaImportNvSciImage(cudaExternalResInterop &cudaExtResObj,
for (int i = 0; i < cudaExtResObj.planeCount; i++) { for (int i = 0; i < cudaExtResObj.planeCount; i++) {
cudaExtent extent = {}; cudaExtent extent = {};
memset(&extent, 0, sizeof(extent)); memset(&extent, 0, sizeof(extent));
extent.width = cudaExtResObj.imageWidth[i]; extent.width = planePitchs[i] / (planePixel[i] / 8);
extent.height = cudaExtResObj.imageHeight[i]; extent.height = planeAlignedHeight[i];
extent.depth = 0; extent.depth = 0;
cudaChannelFormatDesc desc; cudaChannelFormatDesc desc;
switch (channelCount) { switch (channelCount) {