Commit Graph

1331 Commits (master)
 

Author SHA1 Message Date
sherie bfdd1f9c42
[Fix] Fix boxes.scalar_type in nms_npu (#2731) 2023-04-04 16:05:42 +08:00
Zhang bc727f7132
[Enhancement] Repalce the implementation of roiaware_pool3d with mlu-ops. (#2699)
* [Feature] Repalce the implementation of roiaware_pool3d with mlu-ops.

* [Feature] Repalce the implementation of roiaware_pool3d with mlu-ops.
2023-04-03 23:35:07 +08:00
liuduanhui a55f4b7f40
[Enhancement] Replace the implementation of three_nn_forward with mlu-ops (#2719) 2023-04-03 23:30:40 +08:00
sherie f946a933bb
[Enhancement] Add the dtype limit of nms_npu to maintain consistency with the GPU (#2724)
* Increase the dtype limit to maintain consistency with the gpu.

* update error message

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

---------

Co-authored-by: momo609 <963372609.qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-04-03 10:02:39 +08:00
Zaida Zhou 8ceac9347e
Remove .owners.yml (#2712) 2023-03-28 17:06:37 +08:00
bdf 66c5515174
[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>
2023-03-24 19:59:43 +08:00
liuhw 4c4ba6cb17
[Fix] Force bbox_overlaps calculation with FP32 for ascend device (#2697)
* modify bbox_overlaps op adapter

* update
2023-03-24 19:08:57 +08:00
bdf 5f1d55885e
[Enhance] Ignore mlu-ops files (#2691)
Co-authored-by: budefei <budefei@cambricon.com>
2023-03-24 17:00:16 +08:00
bdf 9fcf48c4c8
[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>
2023-03-24 10:27:45 +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
zcc9707 e6f434c440
[Enhance] Add a comment for the use of masked_conv on Ascend device (#2689)
* Update masked_conv.py

* Update mmcv/ops/masked_conv.py

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-22 20:37:15 +08:00
CokeDong 1de3aeffd7
Fix bbox_overlaps of parrots (#2684)
* Fix parrots bbox_overlaps

* Fix
2023-03-22 11:39:55 +08:00
liuhw 4e8972fbf9
Move voxel_size to the device where points are (#2681)
* ...

* ...

* ...
2023-03-20 22:08:30 +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
NoCakeAnyMore a3e97360a9
[Fix] nms_rotated_npu accepts angle parameter (#2670) 2023-03-17 16:40:01 +08:00
sherie 75290685b7
[Enhancement] Refine the implementation of NMS op in Ascend device (#2658) 2023-03-15 23:09:33 +08:00
liuhw 9a671e48fb
[Fix] Prevent divide-by-zero error on Ascend device for bbox_overlaps (#2644) 2023-03-15 23:07:41 +08:00
mahxn0 126077df5e
[Feature] Make voxelization operator support the mlu290 platform (#2652)
* [Feature]: Make voxelization operator support the mlu290 platform.

* [Feature]: Make voxelization operator support the mlu290 platform.

* [Feature]: Make voxelization operator support the mlu290 platform.

---------

Co-authored-by: maxiangjun <maxiangjun@cambricon.com>
2023-03-10 11:21:43 +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
BinZheng 5f122293bf
[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>
2023-03-03 11:18:19 +08:00
jayggh ec63932333
[Enhance] The argument scale of fused_bias_leakyrelu should be greater than 0 (#2623)
* Add verification for scale, when scale less than 0

* Update mmcv/ops/fused_bias_leakyrelu.py

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

* Update fused_bias_leakyrelu.py

* Update fused_bias_leakyrelu.py

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-27 19:03:46 +08:00
BainOuO 4adddfddf0
Fix the paper link error in the docstring (#2618) 2023-02-27 14:53:24 +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
Hakjin Lee cb94ffb672
[Fix] Fix a typo in the docstring of MSDeformAttn (#2599) 2023-02-13 16:59:25 +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
Zaida Zhou 04e8727adf
Remove unused files (#2586) 2023-02-06 12:50:07 +08:00
Zaida Zhou 18e8f32fff
Refactor the issue templates (#2585)
* Refactor the issue template

* fix links

* fix
2023-02-06 11:51:23 +08:00
liuduanhui 46ea829b67
[Fix] Fix tensor descriptor setting in MLU ball_query. (#2579) 2023-02-02 21:44:42 +08:00
Allan Zelener 497c2df495
[Fix] Add missing contiguous calls for nms_rotated (#2547)
Addresses issue https://github.com/open-mmlab/mmcv/issues/2542

For nms_rotated_cuda only adds contiguous call to dets_sorted which is accessed directly.
For nms_rotated_cpu the implementation now matches what's in the detectron2 repo.
2023-02-01 14:44:56 +08:00
bdf 87768b7f2d
[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>
2023-01-31 19:40:07 +08:00
Daojiang 98766f4def
[Docs] Refine the docstring of EvalHook (#2576)
* 📃 docs(hooks): EvalHook comments

Refine the EvalHook comments

* Refine comments EvalHook

Update mmcv/runner/hooks/evaluation.py

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

* 📃 docs(Hooks): update Eval Hook

Slove line too long

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-01-31 17:31:03 +08:00
tripleMu 7e063e3e61
Use official isort (#2574) 2023-01-31 12:59:35 +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
Zaida Zhou db391c50a3
[Fix] Upgrade the version of isort to fix lint error (#2572) 2023-01-30 15:41:55 +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
liuhw 0b005c52b4
[Fix] Fix the parameter error of DeformableRoiPool in Ascend device (#2546) 2023-01-19 21:42:27 +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
Mashiro 4032b83270
Fix a typo (#2549) 2023-01-17 23:33:13 +08:00
mengpenghui e0323b4e4b
[Fix] Check the version of torchvision in __init__ of DCN (#2556) 2023-01-17 19:19:35 +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
mengpenghui c9d477bb27
[Fix] Fix a dilation bug of MLU-DCNv2 and add limitation of torchvision (#2519) 2023-01-12 13:06:22 +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
CokeDong 64e739e002
Update parrots image (#2522)
* Update image

* Bugfix for parrots ext ops

* Move parrots ci task to build_pat.yml

* Fix on:push

* Remove commets and fix

* Fix beautiful job name
2023-01-09 19:41:48 +08:00
yancong 859d84c5b1
[Fix] Fix CI failed due to the version of setuptools (#2530)
* debug build_cu102 ci jobs

* debug build_cu102 ci jobs

* add tmate ssh

* add tmate ssh

* install specified version for setuptools

* revert comments of ci jobs

* revert comments of ci jobs

* specify setuptools version for build_cu116
2023-01-08 14:31:21 +08:00
Mashiro 538bdaacb6
[Fix] Fix c++/cuda lint failed (#2528)
* Fix c/cuda lint failed

* minor refine

* minor refine
2023-01-06 23:03:39 +08:00