diff --git a/pytorch3d/csrc/iou_box3d/iou_utils.cuh b/pytorch3d/csrc/iou_box3d/iou_utils.cuh index 5ad5b165..c4a1f5c9 100644 --- a/pytorch3d/csrc/iou_box3d/iou_utils.cuh +++ b/pytorch3d/csrc/iou_box3d/iou_utils.cuh @@ -728,7 +728,7 @@ __device__ inline int BoxIntersections( } } // Update the face_verts_out tris - num_tris = offset; + num_tris = min(MAX_TRIS, offset); for (int j = 0; j < num_tris; ++j) { face_verts_out[j] = tri_verts_updated[j]; }