mmcv/tests/test_ops
huhongsun 9f43b8c310
add npu roipoint_pool3d_forward (#3147)
* add npu roipoint_pool3d_forward

* nms_rotated npu add new attr is_angle
2024-07-19 01:33:34 +08:00
..
output.pkl [Features] Add stack ball query and stack group points ops (#2292) 2022-10-28 00:42:10 +08:00
test_active_rotated_filter.py [Feature] Add the support of arf op for ascend device (#2792) 2023-05-10 17:07:17 +08:00
test_assign_score_withk.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_ball_query.py [Feature] torch_npu support aclnn and add op (#2998) 2024-01-07 17:11:36 +08:00
test_bbox.py temporarily disable mps ops for torch2.1.0 (#2958) 2023-10-11 13:08:35 +08:00
test_bezier_align.py [Feature] Add BezierAlign CUDA op (#2393) 2023-01-10 17:05:39 +08:00
test_bias_act.py [Feature] Add Ops of StyleGAN3 (#2290) 2023-03-13 16:05:11 +08:00
test_bilinear_grid_sample.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_border_align.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_box_iou_quadri.py [Feature] Add box_iou_quadri & nms_quadri (#2277) 2022-10-13 17:26:14 +08:00
test_box_iou_rotated.py [Feature] Add the support of BoxIouRotated op for ascend device (#2842) 2023-07-19 10:36:26 +08:00
test_carafe.py [Feature] Add carafe op for MLU (#2212) 2022-08-28 17:54:20 +08:00
test_cc_attention.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_chamfer_distance.py [Fix] Fix chamfer_distance cuda bug len should div to 4 (#3113) 2024-05-23 21:17:31 +08:00
test_contour_expand.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_conv_gradfix.py Fix lint (#2702) 2023-03-26 02:06:28 +08:00
test_convex_iou.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_corner_pool.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_correlation.py [Fix] Fix the unit test of correlation op (#1659) 2022-01-13 14:27:50 +08:00
test_deform_conv.py [Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend (#2823) 2023-07-27 16:28:42 +08:00
test_deform_roi_pool.py [Feature] Pick npu ops from master to 2.x (#2501) 2023-01-13 13:39:35 +08:00
test_diff_iou_rotated.py [Feature] Add the implementation of diff_iou_rotated with mlu-ops (#2840) 2023-06-30 15:18:39 +08:00
test_filtered_lrelu.py [Fix] Fix the support for ROCm (#2811) 2023-05-31 12:47:46 +08:00
test_focal_loss.py [Feature] Pick npu ops from master to 2.x (#2501) 2023-01-13 13:39:35 +08:00
test_furthest_point_sample.py Support fps and points_in_box ops for Ascend device (#3085) 2024-04-23 17:31:51 +08:00
test_fused_bias_leakyrelu.py [Feature] Pick npu ops from master to 2.x (#2501) 2023-01-13 13:39:35 +08:00
test_gather_points.py [Feature] Add support of some ops for Ascend device (#2594) 2023-02-13 16:43:42 +08:00
test_group_points.py [Feature] add group_points support npu (#3120) 2024-06-04 14:35:39 +08:00
test_info.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_iou3d.py Support fps and points_in_box ops for Ascend device (#3085) 2024-04-23 17:31:51 +08:00
test_knn.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_masked_conv2d.py [Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend (#2823) 2023-07-27 16:28:42 +08:00
test_merge_cells.py [Enhancement] Lower the restrictions of _resize method in BaseMergeCell (#1959) 2022-05-20 15:13:13 +08:00
test_min_area_polygons.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_modulated_deform_conv.py [Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend (#2823) 2023-07-27 16:28:42 +08:00
test_ms_deformable_attn.py Add multi_scale_deform_attn_grad op adapter for NPU (#3042) 2024-03-13 15:15:46 +08:00
test_nms.py Remove fileio from mmcv and use mmengine.fileio instead (#2179) 2022-08-16 13:37:19 +08:00
test_nms_quadri.py [Feature] Add box_iou_quadri & nms_quadri (#2277) 2022-10-13 17:26:14 +08:00
test_nms_rotated.py Pick MLU modifications from master (1.x) to main (2.x) (#2704) 2023-04-19 10:42:07 +08:00
test_onnx.py [Fix] Fix torch2.0 dcn/mdcn symbolic (#2695) 2023-03-24 13:01:56 +08:00
test_pixel_group.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_points_in_polygons.py [Feature] Add support of points_in_polyogns for Ascend device (#2848) 2023-07-06 15:34:51 +08:00
test_prroi_pool.py [Feature] Support PrRoI op for Parrots (#2280) 2022-10-22 13:39:50 +08:00
test_psa_mask.py [Feature] Pick npu ops from master to 2.x (#2501) 2023-01-13 13:39:35 +08:00
test_riroi_align_rotated.py [Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
test_roi_align.py [Feature] Support mmcv ext with DIOPI impl (#2790) 2023-06-13 17:14:49 +08:00
test_roi_align_rotated.py [Refactor] Repalce the implementation of roi_align_rotated with mlu-ops (#2808) 2023-06-01 00:56:49 +08:00
test_roi_pool.py [Feature] Add support of some ops for Ascend device (#2594) 2023-02-13 16:43:42 +08:00
test_roiaware_pool3d.py Support fps and points_in_box ops for Ascend device (#3085) 2024-04-23 17:31:51 +08:00
test_roipoint_pool3d.py add npu roipoint_pool3d_forward (#3147) 2024-07-19 01:33:34 +08:00
test_rotated_feature_align.py [Feature] Add NPU operator RotatedFeatureAlign (#2994) 2023-11-29 17:24:28 +08:00
test_saconv.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_scatter_points.py [Feature] Add the implementation of dynamic_scatter with mlu-ops (#2847) 2023-06-29 17:13:57 +08:00
test_spconv.py Pick MLU modifications from master (1.x) to main (2.x) (#2704) 2023-04-19 10:42:07 +08:00
test_syncbn.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_three_interpolate.py [Feature] Add the support of three_interpolate op for Ascend device (#2962) 2023-10-17 10:40:09 +08:00
test_three_nn.py [Fix] Fix three nn op can not accept half tensor (#2348) 2022-11-23 16:57:35 +08:00
test_tin_shift.py [Refactor] Move is_mlu_available to mmcv/utils/device_type.py (#1884) 2022-04-17 00:15:35 +08:00
test_upfirdn2d.py [Feature] Add Ops of StyleGAN3 (#2290) 2023-03-13 16:05:11 +08:00
test_voxelization.py [Feature] Support mmcv ext with DIOPI impl (#2790) 2023-06-13 17:14:49 +08:00