[Fix](bangc-ops): fix focalLossSigmoidBackward buffer overflow (Cambr…

…icon#780) Co-authored-by: laijiandong <[email protected]>
tcd-fang · Jul 17, 2023 · db12dce · db12dce
1 parent 881c7f5
commit db12dce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bangc-ops/kernels/focal_loss_sigmoid/focal_loss_sigmoid_backward_union1.mlu b/bangc-ops/kernels/focal_loss_sigmoid/focal_loss_sigmoid_backward_union1.mlu
@@ -257,7 +257,7 @@ __mlu_func__ void focalLossSigmoidBackwardBlock(
   int32_t compute_num = 0;
   int32_t compute_size = 0;
   int32_t compute_align_size = NFU_ALIGN_SIZE;
-  const int32_t nfu_align_num = NFU_ALIGN_SIZE / sizeof(T);
+  const int32_t nfu_align_num = NFU_ALIGN_SIZE / sizeof(float);
   if (sizeof(T) == sizeof(half)) {
     compute_align_size += NFU_ALIGN_SIZE;
   }