mmcv/tests/test_ops
ckirchhoff de6b0021af
[Feature] Support masked_conv in Ascend device (#2387)
* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* clean code

* fix autocast bugs on npu (#2273)

fix autocast bugs on npu (#2273)

* code format

* code format

* code format

* bug fix

* pytorch_npu_helper.hpp clean code

* Npu dev (#2306)

* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather

* raise ImportError when compile with npu

* add npu test case (#2307)

* add npu test case

* Update focal_loss.py

* add comment

* clean lint

* update dtype assert

* update DDP forward and comment

* fix bug

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* sigmoidfocalloss npu adapter bug fix

* BugFix: modify softmaxFocalLoss adapter

* BugFix: remove equal sign in the code

* add npu install information in README

* add modulatedDeformConv npu adapter

* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* add modulatedDeformConv npu adapter

* merge master branch 20221103

* Add masked_ Conv2d operator in NPU

* add nms_npu

* fix bug

* fix code check

* fix code check

* fix code check

* Masked_conv2d NPU

* Masked_conv2d NPU

* Masked_conv2d NPU

* remove npu-install-info in README.md

* annotate the clang-format in pre-commit-config-zh-ch.yaml

* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv

Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-11-15 10:23:17 +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 [Features] Add stack ball query and stack group points ops (#2292) 2022-10-28 16:53:33 +08:00
test_bbox.py [Feature] Add MPS bbox overlap (#2123) 2022-07-22 19:30:01 +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 Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_deform_roi_pool.py [Feature] Support DeformRoiPool with cambricon MLU backend (#2137) 2022-08-28 22:37:28 +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 Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_gather_points.py
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 Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_ms_deformable_attn.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +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 [Fix] Fix batched_nms for rotated box and add type hints for nms.py (#2006) 2022-06-10 19:49:07 +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 with cambricon MLU backend (#2024) 2022-06-08 16:13:13 +08:00
test_riroi_align_rotated.py
test_roi_align.py
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] Support RoiPool with Cambricon MLU backend (#2073) 2022-06-29 11:24:00 +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
test_spconv.py
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
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
test_upfirdn2d.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_voxelization.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00