mmcv/tests/test_ops
Zaida Zhou 5234915c5e
Build macos wheel for torch2.0.0 (#3008)
2023-12-27 10:43:08 +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 [Feature] Add the support of arf op for ascend device (#2789) 2023-05-11 15:47:08 +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] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520) 2023-01-18 14:41:04 +08:00
test_bbox.py Build macos wheel for torch2.0.0 (#3008) 2023-12-27 10:43:08 +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-28 16:53:33 +08:00
test_box_iou_rotated.py [Feature] Add the support of BoxIouRotated op for ascend device (#2854) 2023-07-19 11:32:45 +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 Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 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 [Feature] Add the implementation of diff_iou_rotated with mlu-ops (#2852) 2023-06-30 15:38:02 +08:00
test_focal_loss.py [Feature] Support training on NPU device (#2262) 2022-09-30 21:05:37 +08:00
test_furthest_point_sample.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
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 Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
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 Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
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 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 (#2864) 2023-07-11 14:17:19 +08:00
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 [Refactor] Replace the MLU ops implementation with mlu-ops (#2750) 2023-05-18 11:23:41 +08:00
test_roi_align_rotated.py [Refactor] Repalce the implementation of roi_align_rotated with mlu-ops (#2657) 2023-06-01 00:59:26 +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 [Refactor] Repalce the implementation of rotated_feature_align with mlu_ops (#2659) 2023-06-01 00:58:29 +08:00
test_saconv.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00
test_scatter_points.py [Feature] Add the implementation of dynamic_scatter with mlu-ops (#2853) 2023-06-30 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