473 Commits

Author SHA1 Message Date
Zaida Zhou
5234915c5e
Build macos wheel for torch2.0.0 (#3008) 2023-12-27 10:43:08 +08:00
q.yao
1bb3dea9ad temporarily disable mps ops for torch2.1.0 (#2958) 2023-12-26 20:04:11 +08:00
MrShadowY
8fbacc446d
[Feature] Add the support of BoxIouRotated op for ascend device (#2854) 2023-07-19 11:32:45 +08:00
long11111111
ca60ccb664
[Feature] Add support of points_in_polyogns for Ascend device (#2864) 2023-07-11 14:17:19 +08:00
Danielmic
407c2bd062
[Feature] Add the implementation of dynamic_scatter with mlu-ops (#2853) 2023-06-30 16:59:32 +08:00
Danielmic
83f1b9e0ab
[Feature] Add the implementation of diff_iou_rotated with mlu-ops (#2852) 2023-06-30 15:38:02 +08:00
tudejiang79
58e7c8ce7a
[Refactor] Repalce the implementation of roi_align_rotated with mlu-ops (#2657) 2023-06-01 00:59:26 +08:00
tudejiang79
fd3fbfe197
[Refactor] Repalce the implementation of rotated_feature_align with mlu_ops (#2659) 2023-06-01 00:58:29 +08:00
David de la Iglesia Castro
445910eff6
DvcliveLoggerHook: use by_epoch for determining when to log (#2149) 2023-05-22 15:10:11 +08:00
bdf
8725e683b4
[Refactor] Replace the MLU ops implementation with mlu-ops (#2750) 2023-05-18 11:23:41 +08:00
liuhw
e197effeca
[Feature] Add the support of arf op for ascend device (#2789) 2023-05-11 15:47:08 +08:00
Zaida Zhou
883d339cfb
Replace the ubuntu-18.04 with 22.04 in CI (#2793) 2023-05-11 14:03:24 +08:00
Zhang
595f8fdff3
[Feature] Add the MLU support for box_iou_rotated op (#2703) 2023-04-20 19:31:08 +08:00
liuyuan1-v
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
duzekun
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
sherie
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
liuhw
745985937a
[Feature] Add the support of voxelization op for ascend device (#2613)
* update

* update
2023-03-03 20:07:36 +08:00
Hakjin Lee
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
liuhw
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
xinliang123
615d28176a
[Feature] Add Ascend support for RoIPoolGrad op (#2569)
* add roipoolgrad adapter

* amend
2023-02-06 18:29:59 +08:00
enemy1205
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
ckirchhoff
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
liuduanhui
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
zcc-zjut
84f60c178c
[Feature] Add support for Ascend devices with gather_points (#2555) 2023-01-18 10:33:56 +08:00
mengpenghui
71ee2a61f2
[Enhance] Add AMP support for MLU_DCNv2 (#2548) 2023-01-13 17:51:02 +08:00
mengpenghui
c310d28c8f
[Feature] Add MLU support for DCN (#2540) 2023-01-13 17:46:19 +08:00
sherie
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
ZShaopeng
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
DaGaiBa
73856344f5
[Feature] Support PSAMask op for Ascend device (#2487) 2023-01-06 20:15:50 +08:00
Zhongyu Li
935ba78b39
[Fix] Fix skip_layer for RF-Next (#2489)
* judge skip_layer by fullname

* lint

* skip_layer first

* update unit test
2022-12-28 15:27:34 +08:00
jayggh
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
jayggh
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
Zaida Zhou
4e67e481a0
[Fix] Fix ut caused by the new version of packaging (#2482) 2022-12-11 17:59:11 +08:00
mengpenghui
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
SUNYC
c1712ee290
[Enhance] Add conf_path in PetrelBackend (#2459)
* feat: add conf_path in PetrelBackend

* fix: add conf_path in MockPetrelClient

* fix: adjust parameters order
2022-11-28 11:48:13 +08:00
Shanghua Gao
6e3827d02b
[Feature] Support receptive field search of CNN models (#2056)
* support rfsearch

* add labs for rfsearch

* format

* format

* add docstring and type hints

* clean code

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

* rm unused func

* update code

* update code

* update code

* update  details

* fix details

* support asymmetric kernel

* support asymmetric kernel

* Apply suggestions from code review

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

* Apply suggestions from code review

* Apply suggestions from code review

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

* Apply suggestions from code review

* Apply suggestions from code review

* Apply suggestions from code review

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

* Apply suggestions from code review

* add unit tests for rfsearch

* set device for Conv2dRFSearchOp

* Apply suggestions from code review

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

* remove unused function search_estimate_only

* move unit tests

* Update tests/test_cnn/test_rfsearch/test_operator.py

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

* Apply suggestions from code review

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

* Update mmcv/cnn/rfsearch/operator.py

Co-authored-by: Yue Zhou <592267829@qq.com>

* change logger

* Update mmcv/cnn/rfsearch/operator.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: lzyhha <819814373@qq.com>
Co-authored-by: Zhongyu Li <44114862+lzyhha@users.noreply.github.com>
Co-authored-by: Yue Zhou <592267829@qq.com>
2022-11-22 19:15:55 +08:00
ZShaopeng
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
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
q.yao
6254ebef8d
[Fix] Create Tensor with new_* method to support amp (#2389) 2022-11-08 15:54:34 +08:00
Zhang
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
VVsssssk
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
Yue Zhou
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
takuoko
9709ff3f8c
[Enhancement] Add a new argument define_metric in wandb hook (#2237)
* wandb define_metric

* add test and some fix based on mmengine PR

* fix test

* add summary warnings

* Update mmcv/runner/hooks/logger/wandb.py

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

* Update mmcv/runner/hooks/logger/wandb.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-24 23:13:07 +08:00
cathyzhang222
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
q.yao
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
Ivan Shcheklein
e417035f5d
[Enhancement] Add ability to pass logger instance to frameworks (#2317)
* Add ability to pass logger instance to frameworks

* refine docstring

* Update mmcv/runner/hooks/logger/dvclive.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-12 12:09:53 +08:00
CokeDong
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
q.yao
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
Zaida Zhou
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
wangjiangben-hw
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