mmcv/tests/test_ops
liuyuan1-v 0d1b224fb1
[Feature] Support NmsRotated with cambricon MLU backend (#2643)
* [Feature] Support NmsRotated with cambricon MLU backend

* [Feature] remove foolproofs in nms_rotated_mlu.cpp

* [Feature] fix lint in test_nms_rotated.py

* [Feature] fix kMLU not found in nms_rotated.cpp

* [Feature] modify mlu support in nms.py

* [Feature] modify nms_rotated support in ops.md

* [Feature] modify ops/nms.py
2023-03-23 11:36:32 +08:00
..
output.pkl [Features] Add stack ball query and stack group points ops (#2292) 2022-10-28 16:53:33 +08:00
test_active_rotated_filter.py
test_assign_score_withk.py
test_ball_query.py [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520) 2023-01-18 14:41:04 +08:00
test_bbox.py [Feature] Add Ascend support for bbox_overlaps (#2580) 2023-02-06 18:33:16 +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
test_box_iou_quadri.py [Feature] Add box_iou_quadri & nms_quadri (#2277) 2022-10-28 16:53:33 +08:00
test_box_iou_rotated.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_carafe.py [Feature] Add carafe op for MLU (#2212) 2022-08-19 10:36:26 +08:00
test_cc_attention.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_chamfer_distance.py [Feature] Add ChamferDistance op in gpu (#1933) 2022-06-08 18:12:57 +08:00
test_contour_expand.py
test_convex_iou.py
test_corner_pool.py
test_correlation.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00
test_deform_conv.py [Feature] Add MLU support for DCN (#2540) 2023-01-13 17:46:19 +08:00
test_deform_roi_pool.py [Feature] Add NPU adapter for Deformable RoIPool (#2481) 2022-12-14 15:09:46 +08:00
test_diff_iou_rotated.py
test_focal_loss.py [Feature] Support training on NPU device (#2262) 2022-09-30 21:05:37 +08:00
test_furthest_point_sample.py
test_fused_bias_leakyrelu.py [Feature] Add FusedBiasLeakyRelu npu adapter (#2474) 2022-12-13 11:13:46 +08:00
test_gather_points.py [Feature] Add support for Ascend devices with gather_points (#2555) 2023-01-18 10:33:56 +08:00
test_group_points.py [Fix] Create Tensor with new_* method to support amp (#2389) 2022-11-08 15:54:34 +08:00
test_info.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_iou3d.py [Feature] Support Iou3d with cambricon MLU backend (#2339) 2022-10-24 15:58:58 +08:00
test_knn.py
test_masked_conv2d.py [Feature] Support masked_conv in Ascend device (#2387) 2022-11-15 10:23:17 +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
test_modulated_deform_conv.py [Enhance] Add AMP support for MLU_DCNv2 (#2548) 2023-01-13 17:51:02 +08:00
test_ms_deformable_attn.py [Enhancement] Support MultiScaleDeformableAttention with AMP (#2541) 2023-02-17 19:27:14 +08:00
test_nms.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_nms_quadri.py [Feature] Add box_iou_quadri & nms_quadri (#2277) 2022-10-28 16:53:33 +08:00
test_nms_rotated.py [Feature] Support NmsRotated with cambricon MLU backend (#2643) 2023-03-23 11:36:32 +08:00
test_onnx.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00
test_pixel_group.py
test_points_in_polygons.py
test_prroi_pool.py [Feature] Support PrRoI op for Parrots (#2280) 2022-10-11 15:21:00 +08:00
test_psa_mask.py [Feature] Support PSAMask op for Ascend device (#2487) 2023-01-06 20:15:50 +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] Add Ascend support for RoIAlign op (#2638) 2023-03-20 19:10:42 +08:00
test_roi_align_rotated.py [Feature] Support RoiAlignRotated with cambricon MLU backend (#2033) 2022-06-08 14:13:28 +08:00
test_roi_pool.py [Feature] Add Ascend support for RoIPoolGrad op (#2569) 2023-02-06 18:29:59 +08:00
test_roiaware_pool3d.py [Feature] Support RoIAwarePool3d with cambricon MLU backend (#2359) 2022-11-03 11:13:34 +08:00
test_roipoint_pool3d.py [Feature] Support RoipointPool3d with cambricon MLU backend (#2247) 2022-09-08 17:19:12 +08:00
test_rotated_feature_align.py [Fix] Skip CPU test in test_rotated_feature_align.py for parrots (#1968) 2022-05-20 17:04:58 +08:00
test_saconv.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00
test_scatter_points.py [Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
test_spconv.py [Feature] Add MLU support for Sparse Convolution op (#2589) 2023-03-20 19:28:13 +08:00
test_syncbn.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_tensorrt.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00
test_tensorrt_preprocess.py [Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
test_three_interpolate.py [Fix] Create Tensor with new_* method to support amp (#2389) 2022-11-08 15:54:34 +08:00
test_three_nn.py [Fix] Fix three nn op can not accept half tensor (#2348) 2022-10-24 11:15:21 +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 Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_voxelization.py [Feature] Add the support of voxelization op for ascend device (#2613) 2023-03-03 20:07:36 +08:00