Fix maskrcnn kernel issue on Jetson Nano

Signed-off-by: Tyler Zhu <tylerz@nvidia.com>
NVIDIA · Jul 17, 2020 · 3a68b0a · 3a68b0a
1 parent d9333c3
commit 3a68b0a
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());