Merge pull request #694 from Tyler-D/dev/fix_maskrcnn_kernel_issue

Fix maskrcnn kernel issue on Jetson Nano
NVIDIA · Jul 17, 2020 · 650d4b6 · 650d4b6
2 parents d9333c3 + 3a68b0a
commit 650d4b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugin/common/kernels/maskRCNNKernels.cu b/plugin/common/kernels/maskRCNNKernels.cu
@@ -1732,7 +1732,7 @@ cudaError_t MultilevelPropose(cudaStream_t stream, int N, int inputCnt, int samp
 
     CUASSERT(cudaGetLastError());
 
-    status = KeepTopKGatherBoxScore<1024>(stream, N, dtype, samples, param.keepTopK, sortClassValidCountPtr, sortClassScorePtr,
+    status = KeepTopKGatherBoxScore<512>(stream, N, dtype, samples, param.keepTopK, sortClassValidCountPtr, sortClassScorePtr,
         sortClassLabelPtr, argMaxBBoxPtr, sortClassSampleIdxPtr, sortNMSMarkPtr, outScore, outBbox, 1);
 
     CUASSERT(cudaGetLastError());