Skip to content

Commit

Permalink
[Feature](bangc-ops):Roiaware_pool3d specify 'paramcheck'
Browse files Browse the repository at this point in the history
  • Loading branch information
chqy99 committed Oct 8, 2023
1 parent 2cd2f13 commit 7a61721
Showing 1 changed file with 0 additions and 15 deletions.
15 changes: 0 additions & 15 deletions bangc-ops/kernels/roiaware_pool3d/roiaware_pool3d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -248,16 +248,6 @@ mluOpStatus_t MLUOP_WIN_API mluOpRoiawarePool3dForward(
}
}

/* max_pts_each_voxel affects the allocation of NRAM memory space,
so it's limited by the size of NRAM memory space. */
if (rois_desc->dtype == MLUOP_DTYPE_FLOAT) {
PARAM_CHECK(API, max_pts_each_voxel <=
THRESHOLD_OF_MAX_PTS_EACH_VOXEL_FLOAT_FORWARD);
} else {
PARAM_CHECK(API, max_pts_each_voxel <=
THRESHOLD_OF_MAX_PTS_EACH_VOXEL_HALF_FORWARD);
}

const uint64_t tensor_rois_num = mluOpGetTensorElementNum(rois_desc);
const uint64_t tensor_pts_num = mluOpGetTensorElementNum(pts_desc);
const uint64_t tensor_pts_feature_num =
Expand Down Expand Up @@ -559,11 +549,6 @@ mluOpStatus_t MLUOP_WIN_API mluOpRoiawarePool3dBackward(
return MLUOP_STATUS_BAD_PARAM;
}

/* max_pts_each_voxel affects the allocation of NRAM memory space,
so it's limited by the size of NRAM memory space. */
PARAM_CHECK(API, max_pts_each_voxel <=
THRESHOLD_OF_MAX_PTS_EACH_VOXEL_BACKWARD);

const uint64_t tensor_pts_idx_of_voxels_num =
mluOpGetTensorElementNum(pts_idx_of_voxels_desc);
const uint64_t tensor_argmax_num = mluOpGetTensorElementNum(argmax_desc);
Expand Down

0 comments on commit 7a61721

Please sign in to comment.