36e6b19b78
[Docs] Add npu build.md and update roi_pool_npu adpater ( #2812 )
2023-06-09 10:35:41 +08:00
c9627e867f
[FIx] Fix arf op's write conflict when num_orientations is not 1 ( #2824 )
2023-06-09 10:34:23 +08:00
50d1fffb54
[Fix] Keep the shape of iou op's parameter 2 smaller than parameter 1 ( #2821 )
2023-06-09 10:30:56 +08:00
f7382417d0
[Refactor] Simplify the logic of sparse_conv ( #2802 )
2023-06-01 14:16:20 +08:00
ba8aa76458
[Refactor] Repalce the implementation of roi_align_rotated with mlu-ops ( #2808 )
2023-06-01 00:56:49 +08:00
d2aecbe494
[Feature] Add the support for rotated_feature_align with MLU ( #2809 )
2023-06-01 00:55:20 +08:00
92b3e861a4
[Refactor] Replace the implementation of psa_mask with mlu-ops. ( #2810 )
2023-06-01 00:51:06 +08:00
2611b9900f
[Refactor] Replace carafe op of MLU backend with mlu-ops ( #2817 )
2023-06-01 00:50:04 +08:00
7ff7095ca7
[Fix] Fix the support for ROCm ( #2811 )
2023-05-31 12:47:46 +08:00
3269278ed0
[Fix] Fix type hint in geometric ( #2799 )
2023-05-22 11:06:56 +08:00
0c23eb0215
Sync main with mmcv1.x branch ( #2800 )
2023-05-19 15:40:12 +08:00
59c1418e45
[Enhancement] Make build_xxx_layer allow accepting a class type ( #2782 )
2023-05-11 17:51:13 +08:00
b4dee63ce1
[FIx] Build CUDA ops in CI ( #2794 )
2023-05-10 20:30:19 +08:00
387b63b307
[Enhance] Raise a error if fails to read image ( #2791 )
...
* add assert
* Update mmcv/transforms/loading.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* update comments
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-05-10 20:11:15 +08:00
9a5685a995
[Feature] Add the support of arf op for ascend device ( #2792 )
2023-05-10 17:07:17 +08:00
107f184386
[Fix] ops can not be built in github action ( #2783 )
...
* Fix CI
* Fix CI
* revert comments
* print cl
* use latest
* install wheel
* install wheel
* fix
* Update .github/workflows/pr_stage_test.yml
2023-05-10 16:01:00 +08:00
89a264527e
Enhance issue templates ( #2776 )
2023-04-23 10:55:16 +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
1f161f6859
Remove .owners.yml ( #2743 )
2023-04-07 15:39:25 +08:00
463eab6b5d
[Docs] Describe branch changes ( #2741 )
...
* [Docs] Describe branch changes
* fix typo
* Update English
* refine
2023-04-06 23:42:14 +08:00
7c136e79d5
Update the supported torch and cuda versions in README ( #2687 )
...
* Update the supported torch and cuda versions in README
* update version.json
* use ubuntu-22.04
* refine
2023-04-06 19:22:57 +08:00
c74b488917
Remove two outdated files ( #2740 )
2023-04-06 17:01:01 +08:00
4e768bf3f6
[Fix] Fix parrots CI ( #2737 )
...
* Chery pick from PR #2515 and #2684
* Fix
2023-04-06 16:27:47 +08:00
d429df1a8d
[Refactor] Refactor the pytorch and cuda combination for testing in CI ( #2693 )
...
* Refactor CI
* Upload coverage to Codecov
* update
* update
* update
* update
* test
* test
* fix typo
* fix
* fix
* fix
* fix circleci
* fix circleci
* revert comments
* add verpose
* install cuda
install cuda
install cuda
install cuda
install cuda
install cuda
install cuda
fix env
fix env
fix env
fix env
fix
fix
fix
fix
fix
fix
fix
fix
fix
fix
fix
fix
fix
fix circleci
fix circleci
remove cuda11.1 test from windows
update ci
update ci
update ci
update ci
update ci
update ci
update ci
update ci
* do not test python3.11
* revert comments
2023-04-06 15:01:50 +08:00
1db3967e86
Fix lint ( #2702 )
2023-03-26 02:06:28 +08:00
5b0cdccc5c
[Fix] Enhance the compatibility of training stylegan 2 ( #2694 )
...
* [Fix] Enhance the compatibility of training stylegan 2
* Fix unit test in PyTorch 2.0
* Apply suggestions from code review
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-26 01:58:38 +08:00
aa250d0afa
[Fix] Replace torch/extension.h with torch/types.h to fix compilation error on Windows ( #2698 )
...
* replace extension to types
* update setup
* Update setup.py
* Update setup.py
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-25 11:42:08 +08:00
a6d7dde77f
[Fix] Fix torch2.0 dcn/mdcn symbolic ( #2695 )
...
* fix
* fix lint
2023-03-24 13:01:56 +08:00
5a45fac989
Do not build pre-built pakcages for torch1.6 and 1.7 on macOS platform ( #2678 )
...
* build macos
* fix python version
* fix python version
* Do not build pre-built pakcages for torch1.6 and 1.7 in macOS platform
2023-03-21 13:54:00 +08:00
6a5cd17702
Bump version to v2.0.0 ( #2675 )
2023-03-21 10:23:56 +08:00
c458829763
Refine rfsearch and fix a typo
2023-03-20 18:26:52 +08:00
1f9e5b57c2
[Fix] Fix a typo in the docstring of MSDeformAttn ( #2599 )
2023-03-20 18:26:52 +08:00
b248efa5c3
Fix the paper link error in the docstring ( #2618 )
2023-03-20 18:26:52 +08:00
4ad79591f9
[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-03-20 18:26:52 +08:00
0e14ce2c1d
[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-03-20 18:26:52 +08:00
bc10b5f0c0
[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>
[Fix] Fix skip_layer for RF-Next (#2489 )
* judge skip_layer by fullname
* lint
* skip_layer first
* update unit test
2023-03-20 18:26:52 +08:00
e7adffb9e0
[Fix] Skip filtered_lrelu ut when cuda is less than 10.2 ( #2677 )
2023-03-19 16:17:45 +08:00
d31b22125f
[Fix] nms_rotated_npu accepts angle parameter ( #2674 )
2023-03-17 17:30:50 +08:00
03ea1c9d99
[Fix] Skip fused_lrelu op when gcc is less than 6.0 or cuda is less than 10.2 ( #2671 )
...
* disable filtered_lrelu_op
* fix lint
* add cuda version check
* warning if disable
2023-03-17 17:29:35 +08:00
91ed30dd53
Build macOS pre-built packages for PyTorch2.0 in CI ( #2668 )
...
* Add support for pytorch2.0
* minor fix
* update
2023-03-17 15:40:35 +08:00
3ad02efde2
[Fix] Prevent divide-by-zero error on Ascend device for bbox_overlaps ( #2646 )
2023-03-15 23:08:15 +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
998e459769
[Fix] Call operator __half explicit ( #2608 )
2023-02-18 00:16:34 +08:00
a019d2a3cc
Register Compose to TRANSFORMS ( #2600 )
2023-02-17 15:07:35 +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
8e2b2bf362
[Docs] Add api references in docs ( #2587 )
2023-02-08 20:37:41 +08:00
c84d52cda4
Use official isort ( #2575 )
2023-01-31 12:59:09 +08:00
6cae40a45b
[Fix] Upgrade the version of isort to fix lint error ( #2567 )
2023-01-29 21:06:25 +08:00
sherie
liuhw
duzekun
tudejiang79
liuduanhui
qipengh
xiabo123
jason_w
bdf
Mashiro
Zaida Zhou
Ezra-Yu
ckirchhoff
CokeDong
q.yao
Hakjin Lee
BainOuO
enemy1205
Shanghua Gao
NoCakeAnyMore
Yifei Yang
Qian Zhao
tripleMu