[Fix](mluOpGenerateProposalsV2): fix memcheck bug.

Cambricon · Nov 28, 2024 · 95211f7 · 95211f7
1 parent 273ccdc
commit 95211f7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/kernels/generate_proposals_v2/generate_proposals_v2_nms_utils.h b/kernels/generate_proposals_v2/generate_proposals_v2_nms_utils.h
@@ -329,9 +329,9 @@ __mlu_func__ void nonMaximumSuppress(
   int core_num = 0;
   int nram_save_count = 0;
 
-  limit =
-      (MAX_NRAM_SIZE - NFU_ALIGN_SIZE - nram_save_limit_count * sizeof(T) * 5) /
-      memory_block;
+  limit = (MAX_NRAM_SIZE - NFU_ALIGN_SIZE * sizeof(T) -
+           nram_save_limit_count * sizeof(T) * 5) /
+          memory_block;
   getComputeParams(scores_num, limit, memory_block, sizeof(T), &max_seg_num,
                    &repeat, &remain_num, &core_num, &core_offset);
   // init nram ptr