mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmcv/docs/zh_cn/understand_mmcv/ops.md

67 lines
4.4 KiB
Markdown
Raw Normal View History

[Docs] Reformat op docs (#2052) * [Docs]:update ops.md * fix missing ops
2022-06-15 10:57:29 +08:00
## 算子
[Docs] Build Chinese docs (#1073) * Remove _build directroy in docs_zh_CN * Change utils title to Chinese * Translate documents * Translate documents * Use symbolic link to avoid repeated images * Use symbolic link to avoid repeated images * fix readme.md * update copyright * refactor docs * rename title to Chinese
2021-07-07 13:10:04 +08:00
[Docs] Reformat op docs (#2052) * [Docs]:update ops.md * fix missing ops
2022-06-15 10:57:29 +08:00
MMCV 提供了检测、分割等任务中常用的算子
[Docs] Translate ops documentation (#1349)
2021-09-21 13:54:37 +08:00
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| Device | CPU | CUDA | MLU | MPS | Ascend |
| ---------------------------- | --- | ---- | --- | --- | ------ |
[Feature] Add the support of arf op for ascend device (#2792)
2023-05-10 17:07:17 +08:00
| ActiveRotatedFilter | √ | √ | | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| AssignScoreWithK | | √ | | | |
[Feature] torch_npu support aclnn and add op (#2998)
2024-01-07 17:11:36 +08:00
| BallQuery | | √ | √ | | √ |
[Feature] Add support of some ops for Ascend device (#2594) * [Feature]: add supports of gather_points, nms_rotated, bbox_overlaps for Ascend device * Apply suggestions from code review --------- Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-13 16:43:42 +08:00
| BBoxOverlaps | | √ | √ | √ | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| BorderAlign | | √ | | | |
[Feature] Add the support of BoxIouRotated op for ascend device (#2842) Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-07-19 10:36:26 +08:00
| BoxIouRotated | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| BoxIouQuadri | √ | √ | | | |
| CARAFE | | √ | √ | | |
[Feature] torch_npu support aclnn and add op (#2998)
2024-01-07 17:11:36 +08:00
| ChamferDistance | | √ | | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| CrissCrossAttention | | √ | | | |
| ContourExpand | √ | | | | |
| ConvexIoU | | √ | | | |
| CornerPool | | √ | | | |
| Correlation | | √ | | | |
[Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend (#2823)
2023-07-27 16:28:42 +08:00
| Deformable Convolution v1/v2 | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| Deformable RoIPool | | √ | √ | | √ |
[Feature] Add the implementation of diff_iou_rotated with mlu-ops (#2840)
2023-06-30 15:18:39 +08:00
| DiffIoURotated | | √ | √ | | |
[Feature] Add the implementation of dynamic_scatter with mlu-ops (#2847)
2023-06-29 17:13:57 +08:00
| DynamicScatter | | √ | √ | | |
Support fps and points_in_box ops for Ascend device (#3085)
2024-04-23 17:31:51 +08:00
| FurthestPointSample | | √ | | | √ |
| FurthestPointSampleWithDist | | √ | | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| FusedBiasLeakyrelu | | √ | | | √ |
[Feature] Add support of some ops for Ascend device (#2594) * [Feature]: add supports of gather_points, nms_rotated, bbox_overlaps for Ascend device * Apply suggestions from code review --------- Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-13 16:43:42 +08:00
| GatherPoints | | √ | | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| GroupPoints | | √ | | | |
| Iou3d | | √ | √ | | |
| KNN | | √ | | | |
| MaskedConv | | √ | √ | | √ |
| MergeCells | | √ | | | |
| MinAreaPolygon | | √ | | | |
[Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend (#2823)
2023-07-27 16:28:42 +08:00
| ModulatedDeformConv2d | √ | √ | √ | | √ |
Add multi_scale_deform_attn op adapter for NPU (#3032)
2024-03-01 15:02:23 +08:00
| MultiScaleDeformableAttn | | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| NMS | √ | √ | √ | | √ |
Pick MLU modifications from master (1.x) to main (2.x) (#2704) * [Feature] Support Voxelization with cambricon MLU device (#2500) * [Feature] Support hard_voxelize with cambricon MLU backend * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Enhance] Optimize the performace of ms_deform_attn for MLU device (#2510) * ms_opt * ms_opt * ms_opt * ms_opt * ms_opt * [Feature] ms_deform_attn performance optimization * [Feature] ms_deform_attn performance optimization * [Feature] ms_deform_attn performance optimization * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520) * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. * [Fix] update operator data layout setting. * [Fix] add cxx compile option to avoid symbol conflict. * [Fix] fix lint errors. * [Fix] update ops.md with info of ball_query support by MLU backend. * [Feature] Fix typo. * [Fix] Remove print. * [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env. * [Fix] update MMCV_MLU_OPS_PATH check logic. * [Fix] update error info when failed to download mlu-ops. * [Fix] check mlu-ops version matching info in mmcv. * [Fix] revise wrong filename. * [Fix] remove f.close and re. * [Docs] Steps to compile mmcv-full on MLU machine (#2571) * [Docs] Steps to compile mmcv-full on MLU machine * [Docs] Adjust paragraph order * Update docs/zh_cn/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/zh_cn/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/en/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/en/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Docs] Modify the format --------- Co-authored-by: budefei <budefei@cambricon.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Fix] Fix tensor descriptor setting in MLU ball_query. (#2579) * [Feature] Add MLU support for Sparse Convolution op (#2589) * [Feature] Add sparse convolution MLU API * [Feature] update cpp code style * end-of-file * delete libext.a * code style * update ops.md --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhancement] Replace the implementation of deform_roi_pool with mlu-ops (#2598) * [Feature] Replace the implementation of deform_roi_pool with mlu-ops * [Feature] Modify code --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhancement] ms_deform_attn performance optimization (#2616) * ms_opt_v2 * ms_opt_v2_1 * optimize MultiScaleDeformableAttention ops for MLU * ms_opt_v2_1 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 --------- Co-authored-by: dongchengwei <dongchengwei@cambricon.com> * [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 * [Enhance] Add a default value for MMCV_MLU_ARGS (#2688) * add mlu_args * add mlu_args * Modify the code --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhance] Ignore mlu-ops files (#2691) Co-authored-by: budefei <budefei@cambricon.com> --------- Co-authored-by: ZShaopeng <108382403+ZShaopeng@users.noreply.github.com> Co-authored-by: BinZheng <38182684+Wickyzheng@users.noreply.github.com> Co-authored-by: liuduanhui <103939338+DanieeelLiu@users.noreply.github.com> Co-authored-by: budefei <budefei@cambricon.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: duzekun <108381389+duzekunKTH@users.noreply.github.com> Co-authored-by: dongchengwei <dongchengwei@cambricon.com> Co-authored-by: liuyuan1-v <125547457+liuyuan1-v@users.noreply.github.com>
2023-04-19 10:42:07 +08:00
| NMSRotated | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| NMSQuadri | √ | √ | | | |
| PixelGroup | √ | | | | |
| PointsInBoxes | √ | √ | | | |
| PointsInPolygons | | √ | | | |
| PSAMask | √ | √ | √ | | √ |
[Feature] Add NPU operator RotatedFeatureAlign (#2994)
2023-11-29 17:24:28 +08:00
| RotatedFeatureAlign | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| RoIPointPool3d | | √ | √ | | |
| RoIPool | | √ | √ | | √ |
[Feature] torch_npu support aclnn and add op (#2998)
2024-01-07 17:11:36 +08:00
| RoIAlignRotated | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| RiRoIAlignRotated | | √ | | | |
Pick changes from 1.x branch (#2738)
2023-04-20 19:27:11 +08:00
| RoIAlign | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| RoIAwarePool3d | | √ | √ | | |
| SAConv2d | | √ | | | |
support cpu sigmoid focal loss op
2023-04-18 22:46:49 +08:00
| SigmoidFocalLoss | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| SoftmaxFocalLoss | | √ | | | √ |
| SoftNMS | | √ | | | |
Pick MLU modifications from master (1.x) to main (2.x) (#2704) * [Feature] Support Voxelization with cambricon MLU device (#2500) * [Feature] Support hard_voxelize with cambricon MLU backend * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Enhance] Optimize the performace of ms_deform_attn for MLU device (#2510) * ms_opt * ms_opt * ms_opt * ms_opt * ms_opt * [Feature] ms_deform_attn performance optimization * [Feature] ms_deform_attn performance optimization * [Feature] ms_deform_attn performance optimization * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520) * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. * [Fix] update operator data layout setting. * [Fix] add cxx compile option to avoid symbol conflict. * [Fix] fix lint errors. * [Fix] update ops.md with info of ball_query support by MLU backend. * [Feature] Fix typo. * [Fix] Remove print. * [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env. * [Fix] update MMCV_MLU_OPS_PATH check logic. * [Fix] update error info when failed to download mlu-ops. * [Fix] check mlu-ops version matching info in mmcv. * [Fix] revise wrong filename. * [Fix] remove f.close and re. * [Docs] Steps to compile mmcv-full on MLU machine (#2571) * [Docs] Steps to compile mmcv-full on MLU machine * [Docs] Adjust paragraph order * Update docs/zh_cn/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/zh_cn/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/en/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/en/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Docs] Modify the format --------- Co-authored-by: budefei <budefei@cambricon.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Fix] Fix tensor descriptor setting in MLU ball_query. (#2579) * [Feature] Add MLU support for Sparse Convolution op (#2589) * [Feature] Add sparse convolution MLU API * [Feature] update cpp code style * end-of-file * delete libext.a * code style * update ops.md --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhancement] Replace the implementation of deform_roi_pool with mlu-ops (#2598) * [Feature] Replace the implementation of deform_roi_pool with mlu-ops * [Feature] Modify code --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhancement] ms_deform_attn performance optimization (#2616) * ms_opt_v2 * ms_opt_v2_1 * optimize MultiScaleDeformableAttention ops for MLU * ms_opt_v2_1 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 --------- Co-authored-by: dongchengwei <dongchengwei@cambricon.com> * [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 * [Enhance] Add a default value for MMCV_MLU_ARGS (#2688) * add mlu_args * add mlu_args * Modify the code --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhance] Ignore mlu-ops files (#2691) Co-authored-by: budefei <budefei@cambricon.com> --------- Co-authored-by: ZShaopeng <108382403+ZShaopeng@users.noreply.github.com> Co-authored-by: BinZheng <38182684+Wickyzheng@users.noreply.github.com> Co-authored-by: liuduanhui <103939338+DanieeelLiu@users.noreply.github.com> Co-authored-by: budefei <budefei@cambricon.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: duzekun <108381389+duzekunKTH@users.noreply.github.com> Co-authored-by: dongchengwei <dongchengwei@cambricon.com> Co-authored-by: liuyuan1-v <125547457+liuyuan1-v@users.noreply.github.com>
2023-04-19 10:42:07 +08:00
| Sparse Convolution | | √ | √ | | |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| Synchronized BatchNorm | | √ | | | |
| ThreeInterpolate | | √ | | | |
| ThreeNN | | √ | √ | | |
| TINShift | | √ | √ | | |
| UpFirDn2d | | √ | | | |
Pick MLU modifications from master (1.x) to main (2.x) (#2704) * [Feature] Support Voxelization with cambricon MLU device (#2500) * [Feature] Support hard_voxelize with cambricon MLU backend * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Feature](bangc-ops): add voxelization op * [Enhance] Optimize the performace of ms_deform_attn for MLU device (#2510) * ms_opt * ms_opt * ms_opt * ms_opt * ms_opt * [Feature] ms_deform_attn performance optimization * [Feature] ms_deform_attn performance optimization * [Feature] ms_deform_attn performance optimization * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520) * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. * [Fix] update operator data layout setting. * [Fix] add cxx compile option to avoid symbol conflict. * [Fix] fix lint errors. * [Fix] update ops.md with info of ball_query support by MLU backend. * [Feature] Fix typo. * [Fix] Remove print. * [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env. * [Fix] update MMCV_MLU_OPS_PATH check logic. * [Fix] update error info when failed to download mlu-ops. * [Fix] check mlu-ops version matching info in mmcv. * [Fix] revise wrong filename. * [Fix] remove f.close and re. * [Docs] Steps to compile mmcv-full on MLU machine (#2571) * [Docs] Steps to compile mmcv-full on MLU machine * [Docs] Adjust paragraph order * Update docs/zh_cn/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/zh_cn/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/en/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update docs/en/get_started/build.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Docs] Modify the format --------- Co-authored-by: budefei <budefei@cambricon.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Fix] Fix tensor descriptor setting in MLU ball_query. (#2579) * [Feature] Add MLU support for Sparse Convolution op (#2589) * [Feature] Add sparse convolution MLU API * [Feature] update cpp code style * end-of-file * delete libext.a * code style * update ops.md --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhancement] Replace the implementation of deform_roi_pool with mlu-ops (#2598) * [Feature] Replace the implementation of deform_roi_pool with mlu-ops * [Feature] Modify code --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhancement] ms_deform_attn performance optimization (#2616) * ms_opt_v2 * ms_opt_v2_1 * optimize MultiScaleDeformableAttention ops for MLU * ms_opt_v2_1 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 * [Feature] ms_deform_attn performance optimization V2 --------- Co-authored-by: dongchengwei <dongchengwei@cambricon.com> * [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 * [Enhance] Add a default value for MMCV_MLU_ARGS (#2688) * add mlu_args * add mlu_args * Modify the code --------- Co-authored-by: budefei <budefei@cambricon.com> * [Enhance] Ignore mlu-ops files (#2691) Co-authored-by: budefei <budefei@cambricon.com> --------- Co-authored-by: ZShaopeng <108382403+ZShaopeng@users.noreply.github.com> Co-authored-by: BinZheng <38182684+Wickyzheng@users.noreply.github.com> Co-authored-by: liuduanhui <103939338+DanieeelLiu@users.noreply.github.com> Co-authored-by: budefei <budefei@cambricon.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: duzekun <108381389+duzekunKTH@users.noreply.github.com> Co-authored-by: dongchengwei <dongchengwei@cambricon.com> Co-authored-by: liuyuan1-v <125547457+liuyuan1-v@users.noreply.github.com>
2023-04-19 10:42:07 +08:00
| Voxelization | √ | √ | √ | | √ |
[Feature] Pick npu ops from master to 2.x (#2501) * merge npu ops from master to 2.x * BugFix: fix merge bugs * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition * merge conflicts in ops.md * [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
| PrRoIPool | | √ | | | |
| BezierAlign | √ | √ | | | |
[Feature] Add Ops of StyleGAN3 (#2290) * add bias_act * support bias_act * support filtered_lrelu * support filtered_lrelu and upfirdn2d * support conv2d_gradfix and fix filtered_lrelu * fix lint * fix lint * fix c++ lint * fix part comments * fix lint * rm redundant header * fix upgrade pip * fix as comment * fix c++ lint * fix ci * fix-ut * fix as comments * add grad check * remove redundant template * Update mmcv/ops/bias_act.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add typehint * fix as comment: * complete type hints * fix lint * add test for conv_gradfix * add test for conv_gradfix * fix lint * modify licenses and ops.md * add zh op md * add torch version policy for conv2d_gradfix * fix lint * fix as comments * rename impl * rm redudant function and add ut * fix as comment * fix lint * fix lint * fix as comments * fix lint * fix ut * fix as comment * fix as comment * fix as comment --------- Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-13 16:05:11 +08:00
| BiasAct | | √ | | | |
| FilteredLrelu | | √ | | | |
| Conv2dGradfix | | √ | | | |