NASP/cuda-samples
85
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2026-10-23 21:28:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix #194 and add Large Kernel Parameters Sample

Browse Source
This commit is contained in:
Rob Nertney
2023-05-31 04:43:22 +00:00
parent e612904184
commit 8004ad59ab
20 changed files with 924 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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