From 3a68b0af8c882d94e369d33f6ace186e5eb5d65d Mon Sep 17 00:00:00 2001 From: Tyler Zhu Date: Fri, 17 Jul 2020 14:37:36 +0800 Subject: [PATCH] Fix maskrcnn kernel issue on Jetson Nano Signed-off-by: Tyler Zhu --- plugin/common/kernels/maskRCNNKernels.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugin/common/kernels/maskRCNNKernels.cu b/plugin/common/kernels/maskRCNNKernels.cu index 393cd41f..385d892a 100644 --- 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());