Skip to content

Commit 650d4b6

Browse files
authored
Merge pull request #694 from Tyler-D/dev/fix_maskrcnn_kernel_issue
Fix maskrcnn kernel issue on Jetson Nano
2 parents d9333c3 + 3a68b0a commit 650d4b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

plugin/common/kernels/maskRCNNKernels.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1732,7 +1732,7 @@ cudaError_t MultilevelPropose(cudaStream_t stream, int N, int inputCnt, int samp
17321732

17331733
CUASSERT(cudaGetLastError());
17341734

1735-
status = KeepTopKGatherBoxScore<1024>(stream, N, dtype, samples, param.keepTopK, sortClassValidCountPtr, sortClassScorePtr,
1735+
status = KeepTopKGatherBoxScore<512>(stream, N, dtype, samples, param.keepTopK, sortClassValidCountPtr, sortClassScorePtr,
17361736
sortClassLabelPtr, argMaxBBoxPtr, sortClassSampleIdxPtr, sortNMSMarkPtr, outScore, outBbox, 1);
17371737

17381738
CUASSERT(cudaGetLastError());

0 commit comments

Comments
 (0)