5234915c5e
Build macos wheel for torch2.0.0 ( #3008 )
2023-12-27 10:43:08 +08:00
1bb3dea9ad
temporarily disable mps ops for torch2.1.0 ( #2958 )
2023-12-26 20:04:11 +08:00
8fbacc446d
[Feature] Add the support of BoxIouRotated op for ascend device ( #2854 )
2023-07-19 11:32:45 +08:00
ca60ccb664
[Feature] Add support of points_in_polyogns for Ascend device ( #2864 )
2023-07-11 14:17:19 +08:00
407c2bd062
[Feature] Add the implementation of dynamic_scatter with mlu-ops ( #2853 )
2023-06-30 16:59:32 +08:00
83f1b9e0ab
[Feature] Add the implementation of diff_iou_rotated with mlu-ops ( #2852 )
2023-06-30 15:38:02 +08:00
58e7c8ce7a
[Refactor] Repalce the implementation of roi_align_rotated with mlu-ops ( #2657 )
2023-06-01 00:59:26 +08:00
fd3fbfe197
[Refactor] Repalce the implementation of rotated_feature_align with mlu_ops ( #2659 )
2023-06-01 00:58:29 +08:00
8725e683b4
[Refactor] Replace the MLU ops implementation with mlu-ops ( #2750 )
2023-05-18 11:23:41 +08:00
e197effeca
[Feature] Add the support of arf op for ascend device ( #2789 )
2023-05-11 15:47:08 +08:00
595f8fdff3
[Feature] Add the MLU support for box_iou_rotated op ( #2703 )
2023-04-20 19:31:08 +08:00
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
06fa32853b
[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>
2023-03-20 19:28:13 +08:00
01a0f53ea4
[Feature] Add Ascend support for RoIAlign op ( #2638 )
...
* add roi_algin op
* add unit test
* update ops.md
* fix typo
---------
Co-authored-by: momo609 <963372609.qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-20 19:10:42 +08:00
745985937a
[Feature] Add the support of voxelization op for ascend device ( #2613 )
...
* update
* update
2023-03-03 20:07:36 +08:00
8e8ab22686
[Enhancement] Support MultiScaleDeformableAttention with AMP ( #2541 )
...
* [Enhance] Support FP16 for MSDeformAttn
* [Fix] Data type mismatch
* Update mmcv/ops/multi_scale_deform_attn.py
* Add UT
Author: nijkah <nijkah@gmail.com>
* Add cuda available condition
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-17 19:27:14 +08:00
5a61e534ea
[Feature] Add Ascend support for bbox_overlaps ( #2580 )
...
* Add support for Ascend devices with bbox_overlaps
* ...
* ...
2023-02-06 18:33:16 +08:00
615d28176a
[Feature] Add Ascend support for RoIPoolGrad op ( #2569 )
...
* add roipoolgrad adapter
* amend
2023-02-06 18:29:59 +08:00
422816e45c
[Feature] Add C++ implementation for bbox_overlaps ( #2477 )
...
* add ops bbox_overlaps
* format code
* Return the pytorch version
* Intermediate modification
* Solve problems in parameter passing
* revise bug
* "add test case"
2023-01-31 00:04:35 +08:00
54ed0ed869
[Feature] Add support for Ascend devices with nms_rotated ( #2550 )
...
* [Feature]: add nms_rotated npu adaptater code
* [BugFix]: modify param in nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated.cpp
* [Doc]: add nms_rotated op in supported op list at ops.md
* [Test]: add nms_rotated unit_test
* [Bug]: remove device parameter in test_batched_nms function
2023-01-28 21:08:22 +08:00
dfb03806a1
[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.
2023-01-18 14:41:04 +08:00
84f60c178c
[Feature] Add support for Ascend devices with gather_points ( #2555 )
2023-01-18 10:33:56 +08:00
71ee2a61f2
[Enhance] Add AMP support for MLU_DCNv2 ( #2548 )
2023-01-13 17:51:02 +08:00
c310d28c8f
[Feature] Add MLU support for DCN ( #2540 )
2023-01-13 17:46:19 +08:00
2810718a99
[Feature] Add Ascend support for RoIPool op ( #2483 )
...
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
2023-01-12 11:52:28 +08:00
48ea88ab9f
[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
2023-01-10 19:43:45 +08:00
73856344f5
[Feature] Support PSAMask op for Ascend device ( #2487 )
2023-01-06 20:15:50 +08:00
46eb9ec5d0
[Feature] Add NPU adapter for Deformable RoIPool ( #2481 )
...
* add npu adapter for deformroipool
* cleancode
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
2022-12-14 15:09:46 +08:00
4c51afce2a
[Feature] Add FusedBiasLeakyRelu npu adapter ( #2474 )
...
* 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
* Create fused_bias_leakyrelu_npu.cpp
Add NPU adapter for fused_bias_leaky_relu operator
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update ops.md
* Update ops.md
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update ops.md
* amend for CI
* bugfix
* amend ops.md
* Update test_fused_bias_leakyrelu.py
* clean code
* bugfix
* clean code
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: ckirchhoff <515629648@qq.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-12-13 11:13:46 +08:00
49d891754b
[Feature] Support modulated_deform_conv with cambricon MLU backend ( #2411 )
...
* [Feature] Support modulated_deform_conv with cambricon MLU backend
* fix error of torch_mlu
* modify with commit suggestion
* Update modulated_deform_conv.py
* Update mmcv/ops/modulated_deform_conv.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-04 17:46:44 +08:00
12442667ff
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend ( #2396 )
...
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support msdeformattn_1104 with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* Revert "[Feature] Support MsDeformAttn with cambricon MLU backend"
This reverts commit 27963cccc86d240852a40a2c1510147a3e9f269f.
* [Feature] Support MsdeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsdeformAttn with cambricon MLU backend
Co-authored-by: zhangshaopeng <wicky-zheng@outlook.com>
Co-authored-by: wicky-zheng <root@notebook-mmcv-290m8-torch1-6-y0yv66-notebook-0.notebook-mmcv-290m8-torch1-6-y0yv66.ns-ad8b689a0ecd41fcb4469c803dcd539d.svc.cluster.local>
2022-11-16 14:08:04 +08:00
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
6254ebef8d
[Fix] Create Tensor with new_* method to support amp ( #2389 )
2022-11-08 15:54:34 +08:00
b622fb2e29
[Feature] Support RoIAwarePool3d with cambricon MLU backend ( #2359 )
...
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-03 11:13:34 +08:00
93fe4829f7
[Features] Add stack ball query and stack group points ops ( #2292 )
...
* add stack sa model ops
* fix lint
* fix lint
* fix comments
* fix bug
* fix lint
* fix comments
* fix lint
* fix lint
* fix
2022-10-28 16:53:33 +08:00
a4f304a5f5
[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-28 16:53:33 +08:00
7fd7058a9c
[Feature] Support Iou3d with cambricon MLU backend ( #2339 )
...
* [Feature] Support Iou3d with cambricon MLU backend
* [Feature] add double line
* [Feature] add line
2022-10-24 15:58:58 +08:00
6cf9ac0507
[Fix] Fix three nn op can not accept half tensor ( #2348 )
...
* Fix three nn half inpt
* update test
2022-10-24 11:15:21 +08:00
dd5b415d2a
[Feature] Support PrRoI op for Parrots ( #2280 )
...
* Support parrots extension for op PrRoI
* Fix lint
* Fix cpp lint
* Fix testcase failure by false requires_grad in self-defined autograd Funtion
* Fix issues
* Fix flake8
* Fix isort
* Adaption for typechecking for PrRoIPoolFunction
* Fix lint
* Support only float32
* bugfix
2022-10-11 15:21:00 +08:00
c001e2fcba
[Refactor] Refactor the nms3d op to support MLU ( #2296 )
...
* update iou3d
* fix parrot
* update to device
* replace count_nonzero with fill
* update build.yml
2022-10-11 10:24:46 +08:00
b6a7fd98e4
Upgrade pre commit hooks ( #2321 )
...
* Upgrade the versions of pre-commit hooks
* update the versions of zh-cn.yaml
2022-10-08 11:48:44 +08:00
a5db5f6682
[Feature] Support training on NPU device ( #2262 )
...
* 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>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-30 21:05:37 +08:00
83bedf5a2b
[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-09-08 17:19:12 +08:00
e843d73fc7
[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-08-28 22:37:28 +08:00
a08517790d
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-23 15:18:47 +08:00
e663670a74
[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-21 23:21:54 +08:00
832b845257
[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-19 10:36:26 +08:00
4558bfbd5b
[Fix] Fix onnx unit tests ( #2155 )
2022-07-29 19:23:29 +08:00
35affc35bf
[Feature] Support PrRoIPool operation ( #2113 )
...
* support prroipooling
* fix build bug
* fix bug of precision
* fix lint
* add ut
* add license and citation
* polish op and add type hint
* refactor ut
* add docstring for prroipool
* polish code
* format cu and cuh code
* fix format
* fix format
* fix typo
Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-07-23 23:08:32 +08:00
22fadceecd
[Feature] Add MPS bbox overlap ( #2123 )
...
* add mps bbox overlap
* format
* update document and manifest
* update readme
2022-07-22 19:30:01 +08:00
Zaida Zhou
q.yao
MrShadowY
long11111111
Danielmic
tudejiang79
bdf
liuhw
Zhang
liuyuan1-v
duzekun
sherie
Hakjin Lee
xinliang123
enemy1205
ckirchhoff
liuduanhui
zcc-zjut
mengpenghui
ZShaopeng
DaGaiBa
jayggh
q.yao
VVsssssk
Yue Zhou
cathyzhang222
CokeDong
wangjiangben-hw
xcnick
Jingwei Zhang