Commit Graph

1365 Commits (5234915c5efda6d080223be6ff6606874876e901)
 

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
q.yao 62d7092a4c Support pre released pytorch2.1.0 (#2865) 2023-12-26 20:04:11 +08:00
Zaida Zhou ed4303ea95
Update version info (#2981) 2023-11-02 11:54:50 +08:00
sherie e5d9d8e106
Fix torch 1.8 adpater npu bug (#2970) 2023-10-20 16:52:14 +08:00
sherie 50d1eaaf23
Support torch_npu v2.1 (#2943) 2023-09-26 16:44:36 +08:00
Zaida Zhou bc01e96d54
[Fix] Delete yapf verify (#2944) 2023-09-26 10:49:53 +08:00
Chris Jiang 43c213037f
[Fix] Update mlu-ops version (#2922) 2023-09-08 10:43:36 +08:00
Chris Jiang 2491dbb7d0
[Refactor] Replace tin_shift op of MLU backend with mlu-ops (#2911) 2023-08-28 17:02:51 +08:00
Chris Jiang a0a17050f2
[Refactor] Replace roipoint_pool3d op of MLU backend with mlu-ops (#2876) 2023-08-28 15:50:35 +08:00
qirun-uiuc 0c4ce221f3
[Refactor] Replace focal_loss_sigmoid op of MLU backend with mlu-ops (#2877) 2023-08-28 15:46:55 +08:00
qirun-uiuc 8cf8a5cdba
[Refactor] Replace bbox_overlaps op of MLU backend with mlu-ops (#2880) 2023-08-28 15:45:12 +08:00
Chris Jiang eb1dad9ad8
[Fix] Fix MLU code format (#2886) 2023-08-07 15:21:49 +08:00
Chris Jiang ea3e3e6032
[Enhancement] Add torch mluops check before call mluOpsxxx interface (#2874) 2023-08-03 16:46:16 +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
Yinlei Sun f07ed6bcc0
[Fix] Fix deform_conv ops on Ascend NPU (#2805) 2023-06-15 16:46:07 +08:00
liuhw 9c0f9cb98c
Revert "[Fix] Fix arf op's write conflict when num_orientations is not 1 (#2822)" (#2826) 2023-06-10 23:12:28 +08:00
liuhw e43982597c
[Fix] Fix arf op's write conflict when num_orientations is not 1 (#2822) 2023-06-09 10:32:39 +08:00
liuhw 16d52da1e0
[Fix] keep the shape of iou op's parameter 2 smaller than parameter 1 (#2820) 2023-06-09 10:30:12 +08:00
duzekun e5dcfba13e
[Refactor] Simplify the logic of sparse_conv (#2683) 2023-06-01 14:16:51 +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
liuduanhui 515d5416a0
[Refactor] Replace the implementation of psa_mask with mlu-ops. (#2756) 2023-06-01 00:57:30 +08:00
qipengh 8ca930cc72
[Refactor] Replace carafe op of MLU backend with mlu-ops (#2818) 2023-06-01 00:48:48 +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
luomaoling 8d01bf6a6c
[Fix] Fix torch_npu may not contain optim (#2766)
* fix npu optimizer

* fix npu optimizer
2023-04-19 10:17:18 +08:00
luomaoling 81163d5cb1
[Feature] Add torch_npu optimizers (#2753)
* add npu optimizer

* add npu optimizer

* add npu optimizer
2023-04-13 23:08:17 +08:00
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