c0e91e90e6
Merge 119fe2cbc9 into f19d3e771c
2025-04-14 11:52:35 +00:00
6a1646f1a3
Support fps and points_in_box ops for Ascend device ( #3085 )
2024-04-23 17:31:51 +08:00
cd05d71254
Add multi_scale_deform_attn op adapter for NPU ( #3032 )
2024-03-01 15:02:23 +08:00
c7c02a7c5b
[Feature] torch_npu support aclnn and add op ( #2998 )
2024-01-07 17:11:36 +08:00
2e44eaeba3
[Feature] Add NPU operator RotatedFeatureAlign ( #2994 )
2023-11-29 17:24:28 +08:00
86a38aa3f0
[Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend ( #2823 )
2023-07-27 16:28:42 +08:00
987d34b0cf
[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
d28aa8a9cc
[Feature] Add the implementation of diff_iou_rotated with mlu-ops ( #2840 )
2023-06-30 15:18:39 +08:00
10c8b9e78b
[Feature] Add the implementation of dynamic_scatter with mlu-ops ( #2847 )
2023-06-29 17:13:57 +08:00
d2aecbe494
[Feature] Add the support for rotated_feature_align with MLU ( #2809 )
2023-06-01 00:55:20 +08:00
0c23eb0215
Sync main with mmcv1.x branch ( #2800 )
2023-05-19 15:40:12 +08:00
9a5685a995
[Feature] Add the support of arf op for ascend device ( #2792 )
2023-05-10 17:07:17 +08:00
d493a5d523
Pick changes from 1.x branch ( #2738 )
2023-04-20 19:27:11 +08:00
733e6ff84e
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
119fe2cbc9
support cpu sigmoid focal loss op
2023-04-18 23:43:59 +08:00
869dbf1bf2
[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
0fb07d0e69
[Feature] Add the support of voxelization op for ascend device ( #2614 )
...
* Add voxelization op npu adaptor
* update
* update
2023-03-03 20:07:11 +08:00
834f94db23
[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
34bdf448ac
[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
7156604eba
[Feature] Add BezierAlign CUDA op ( #2393 )
...
* bezier align
* add ut
* fix comment
* updata ut
* fix link and comment
* fix comment
2023-01-10 17:05:39 +08:00
a0939977a3
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend
2022-11-23 16:57:35 +08:00
6c89b71756
[Feature] Support Iou3d with cambricon MLU backend ( #2339 )
...
* [Feature] Support Iou3d with cambricon MLU backend
* [Feature] add double line
* [Feature] add line
2022-11-23 16:57:35 +08:00
e9777adbac
[Feature] Support RoIAwarePool3d with cambricon MLU backend ( #2359 )
...
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-23 16:57:35 +08:00
a8f7ae48e2
[Feature] Support RoipointPool3d with cambricon MLU backend ( #2247 )
...
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
2022-10-22 13:39:50 +08:00
a364e6cad2
[Feature] Support DeformRoiPool with cambricon MLU backend ( #2137 )
...
* [Feature] Support DeformRoiPool with cambricon MLU backend
* [Fix] Remove use of std library
* [Fix] Correct the error information
* [Refactor] Refactor test deform_roi_pool code
* [Fix] Fix judgment error
* [Fix] Modify the large tensor check
Co-authored-by: budefei <budefei@cambricon.com>
2022-10-22 13:39:50 +08:00
a4c82617ff
[Feature] Add box_iou_quadri & nms_quadri ( #2277 )
...
* update
* update
* fix link
* fix bug
* update nms_quadri
* fix lint
* Update test_nms_quadri.py
* Update box_iou_quadri.py
* fix bug
* Update test_nms_quadri.py
* Update box_iou_rotated_utils.hpp
* Update box_iou_quadri.py
* Update mmcv/ops/nms.py
2022-10-13 17:26:14 +08:00
cdfbdc0bd1
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-28 17:54:20 +08:00
625e82ce62
[Feature] Add carafe op for MLU ( #2212 )
...
* [Feature] Support CARAFE with Cambricon MLU backend
* [Docs] Add comments for common funtions
* [Test] Add allclose test for carafe
* Remove print
Co-authored-by: zcyKTH <zcy19950525@gmail.com>
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 17:54:20 +08:00
f5a19ef06f
[Feature] Support MaskedConv2d with cambricon MLU backend ( #2202 )
...
* [Feature] Support MaskedConv2d with cambricon MLU backend
* [Refactor] Refactor test masked_conv2d code
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 17:54:20 +08:00
0b4285d959
Pick commits from master ( #2164 )
...
* [Docs] Add swith_language.md in docs (#2160 )
* [Fix] Fix onnx unit tests (#2155 )
* [Docs] Limit extension versions (#2144 )
* Support PrRoIPool operation
* Add MPS bbox overlap
* Add .pre-commit-config-zh-cn.yaml (#2135 )
Co-authored-by: xcnick <xcnick0412@gmail.com>
Co-authored-by: Jingwei Zhang <zjw18@mails.tsinghua.edu.cn>
Co-authored-by: q.yao <yaoqian@sensetime.com>
2022-08-03 14:11:53 +08:00
2708fac6c5
[Feature] Support RoiPool with Cambricon MLU backend ( #2073 )
...
* [Feature] Support RoiPool with cambricon MLU backend
* [Docs] Update ops.md
2022-06-29 11:24:00 +08:00
9110df94fc
[Docs] Reformat op docs ( #2052 )
...
* [Docs]:update ops.md
* fix missing ops
2022-06-15 10:57:29 +08:00
834d5978df
[Feature] Add ChamferDistance op in gpu ( #1933 )
...
* update
* fix lint
* fix lint
* Update mmcv/ops/chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update chamfer_distance.py
* Update test_chamfer_distance.py
* fix
* Update chamfer_distance_cuda_kernel.cuh
* Update chamfer_distance_cuda_kernel.cuh
* fix
* Update mmcv/ops/chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-08 18:12:57 +08:00
e9f48a4f8e
[Enhance] Replace BEV IoU with 3D IoU ( #1902 )
...
* add iou3d
* revert deprecated python function
* fix lint
* replace 3d iou/nms calls for bev iou/nms
2022-05-25 17:05:50 +08:00
aee596d523
[Enhance] Differentiable rotated IoU ( #1854 )
...
* diff_iou_rotated is working
* add test; fix lint
* fix lint for test
* disable cpu build
* refactor files structure
* fix comments
* remove extra .repeat()
* add comment
* fix j-1 bug; update doc
* fix clang lint
* update docstrings
* fix comments
* fix comments
2022-04-15 15:40:07 +08:00
c1de4c9bbf
[Feature] Add spconv ops from mmdet3d ( #1581 )
...
* add ops (spconv) of mmdet3d
* fix typo
* refactor code
* resolve comments in #1452
* fix compile error
* fix bugs
* fix bug
* transform from 'types.h' to 'extension.h'
* fix bug
* transform from 'types.h' to 'extension.h' in parrots
* add extension.h in pybind.cpp
* add unittest
* Recover code
* (1) Remove prettyprint.h
(2) Switch `T` to `scalar_t`
(3) Remove useless lines
(4) Refine example in docstring of sparse_modules.py
* (1) rename from `cu.h` to `cuh`
(2) remove useless files
(3) move cpu files to `pytorch/cpu`
* reorganize files
* Add docstring for sparse_functional.py
* use dispatcher
* remove template
* use dispatch in cuda ops
* resolve Segmentation fault
* remove useless files
* fix lint
* fix lint
* fix lint
* fix unittest in test_build_layers.py
* add tensorview into include_dirs when compiling
* recover all deleted files
* fix lint and comments
* recover setup.py
* replace tv::GPU as tv::TorchGPU & support device guard
* fix lint
Co-authored-by: hdc <hudingchang.vendor@sensetime.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
2022-02-18 15:53:10 +08:00
594ff3c0d3
[Feature] Add convex IoU CUDA op for rotated detection. ( #1610 )
...
* add convex iou
* fix lint
* add convex_iou
* fix convex_iou
* add convex_giou
* fix bug
* fix lint
* fix bug
* Update
* update
* add kernel loop
* fix bug.
* fix polygen typo
* simplify reverse
* Update convex_iou_cuda_kernel.cuh
* Update mmcv/ops/convex_iou.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/convex_iou.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add AT_DISPATCH_FLOATING_TYPES_AND_HALF
* fix lint
* fix lint
* Resolving conflicts
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 20:43:12 +08:00
9acc892a44
[Feature] Add active rotated filter op for rotated detection. ( #1598 )
...
* add active_rotated_filter
* fix lint
* fix lint
* renaming nRotation and nOrientation
* Update active_rotated_filter_cuda_kernel.cuh
* Update active_rotated_filter_cuda.cu
* fix bug
* fix lint
* Update test_active_rotated_filter.py
* fix lint
* Update active_rotated_filter_cuda_kernel.cuh
* renaming
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix lint
* Resolving conflicts
* fix lint.
* Update __init__.py
* Update mmcv/ops/csrc/pytorch/cuda/active_rotated_filter_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update active_rotated_filter.cpp
* fix lint
* Update mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 17:35:20 +08:00
51b40c332a
[Feature] Add min_area_polygons CUDA op for rotated detection. ( #1611 )
...
* init
* Update pybind.cpp
* Update min_area_polygons_cuda.cuh
* Update cudabind.cpp
* fix bug
* Create test_min_area_polygons.py
* add test
* update
* Update min_area_polygons_cuda.cuh
* fix bugs.
* Update min_area_polygons_cuda.cuh
* Update min_area_polygons.py
* Update min_area_polygons_cuda.cuh
* merge these 4 nested loops
* add AT_DISPATCH_FLOATING_TYPES_AND_HALF
* fix lint
* Resolving conflicts
2022-01-10 11:00:50 +08:00
304efbb650
[Feature] Add points_in_polygons CUDA op for rotated detection. ( #1600 )
2021-12-24 10:56:48 +08:00
0bcbeadb53
[Feature] Add RiRoIAlignRotated CUDA op for rotated detection. ( #1599 )
2021-12-24 00:30:34 +08:00
2475dc3452
[Feature] Add feature refine CUDA op for rotated detection. ( #1603 )
...
* re PR
* replace all the feature_refine with rotated_feature_align
* replace all the FR with RotatedFeatureAlign
* Update mmcv/ops/rotated_feature_align.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/rotated_feature_align.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix lint
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/cudabind.cpp
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix lint
* Update cudabind.cpp
* Update cudabind.cpp
* fix bug in test. & add backward test
* fix lint
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-12-23 17:38:58 +08:00
e4b5348ebf
[Docs] Refactor the structure of documentation ( #1580 )
...
* [Docs] Refactor the structure of documentation
* [Docs] Refactor the structure of documentation
* fix symlink
* fix link
* fix typo
* polish docstring
* fix docstring
2021-12-15 17:01:09 +08:00
萝卜菜
sherie
DaGaiBa
huaweiZJX
qipengh
MrShadowY
Danielmic
tudejiang79
bdf
liuhw
ckirchhoff
萝卜菜
Yifei Yang
liukuikun
ZShaopeng
cathyzhang222
Zhang
Yue Zhou
liuduanhui
Zaida Zhou
zhouchenyang
Alex Yang
Danila Rukhovich
Wenhao Wu