Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix npu op bug. #3122

Open
wants to merge 73 commits into
base: main
Choose a base branch
from
Open
Changes from 2 commits
Commits
Show all changes
73 commits
Select commit Hold shift + click to select a range
7534dd6
fix npu bug.
momo609 May 13, 2024
ef10dc7
modify chamfer
Annarine Jun 19, 2024
46182c6
Add NPU support for dynamic voxelization
Pr0Wh1teGivee Jun 18, 2024
ee79718
Merge pull request #37 from Pr0Wh1teGivee/rc4main
momo609 Jun 19, 2024
b238850
Bugfix of NPU adapter of nms3d
DaGaiBa Jun 19, 2024
14f031a
repair nms_rotated bug
huhongsun Jun 18, 2024
ffde0d5
Merge pull request #31 from huhongsun/rc4main
momo609 Jun 20, 2024
ca0e427
add dtype check for roi_align
mikaelchan Jun 21, 2024
4aacf0f
Merge pull request #39 from mikaelchan/rc4main
momo609 Jun 21, 2024
2ed5eb5
adapt box_iou_rotaed to boxes_overlap_bev
Jun 21, 2024
d11a531
Merge pull request #38 from DaGaiBa/rc4main
momo609 Jun 22, 2024
a10cfa4
Merge pull request #41 from Pr0Wh1teGivee/iou_fix
momo609 Jun 24, 2024
7c56ba9
fix the bug of DeformableRoiPoolGrad
wujiadi1 Jun 25, 2024
e112ca4
Merge pull request #43 from wujiadi1/rc4main
momo609 Jun 25, 2024
f572475
Interfaces change.
DaGaiBa Jul 19, 2024
44afa6a
Merge pull request #51 from DaGaiBa/rc4main
momo609 Jul 19, 2024
6728a04
chamfer push_back
Annarine Jul 22, 2024
5e0bd89
adapt boxes_overlap_bev & box_iou_rotated
OlaWod Jul 24, 2024
a6d4f0d
fix msda
Pr0Wh1teGivee Aug 5, 2024
0c19b5d
Merge pull request #54 from Pr0Wh1teGivee/msda_fix
momo609 Aug 5, 2024
b7702bf
fix focallossgrad
momo609 Aug 7, 2024
d5085da
Merge pull request #53 from OlaWod/rc4main
momo609 Sep 10, 2024
650c321
points_in_boxes_all的mmcv适配
ZrBac Oct 24, 2024
8c45aa9
Merge pull request #60 from ZrBac/box
hust17yixuan Oct 29, 2024
8f5a5b9
change the knn and three nn code
huangyuan64 Oct 29, 2024
87f00a6
Merge pull request #61 from huangyuan64/rc4main
hust17yixuan Oct 29, 2024
993979b
codeclean npu/boxes_overlap_bev_npu.cpp
OlaWod Oct 30, 2024
94d6e92
adapt npu box_iou_quadri
OlaWod Sep 24, 2024
5260330
Merge pull request #62 from OlaWod/rc4main
hust17yixuan Oct 31, 2024
519195c
Merge pull request #63 from OlaWod/quadri
hust17yixuan Oct 31, 2024
ae66b32
add the roi align rotated v2 ops
huangyuan64 Nov 1, 2024
c29c607
Merge pull request #64 from huangyuan64/rc4
hust17yixuan Nov 1, 2024
ffe3a1a
add assign_score_withk NPU adaptation
Hua-yuxiu Nov 4, 2024
42d93d4
update point_to_voxel & voxel_to_point in scatter_points.py
OlaWod Nov 4, 2024
9634af3
Merge pull request #67 from OlaWod/scatter_points
hust17yixuan Nov 4, 2024
1087e50
Merge pull request #66 from Hua-yuxiu/rc4main_my
hust17yixuan Nov 4, 2024
0535385
border_align
frh23333 Nov 4, 2024
6c48402
Update points_in_boxes.py
ZrBac Nov 5, 2024
96c4d7c
主仓同步
Nov 5, 2024
e3cf445
Merge pull request #70 from ZrBac/patch-2
hust17yixuan Nov 6, 2024
3424ec1
add new npu op roiaware_pool3d
Nov 6, 2024
32997dd
Merge pull request #69 from fangruohuan/rc4main
hust17yixuan Nov 6, 2024
0238175
Merge pull request #73 from JYYCaN/rc4main
hust17yixuan Nov 6, 2024
4a21387
pixel_grou
Annarine Nov 6, 2024
7ee5a41
Merge pull request #76 from Annarine/rc4main
hust17yixuan Nov 6, 2024
eedad49
scatter points bug fix
JYYCaN Nov 7, 2024
1c2c238
Merge pull request #77 from JYYCaN/patch-1
hust17yixuan Nov 7, 2024
7fd21ee
update nms_rotated from openmmlab.mmcv main
abdu-uy Nov 23, 2024
2b7def6
Merge pull request #79 from abdu-uy/rc4main
hust17yixuan Nov 23, 2024
ca26c89
roi_align_rotated_v2
ason-rob Nov 26, 2024
f27fff7
Merge pull request #81 from ason-rob/rc4main
hust17yixuan Nov 27, 2024
cf23718
add pixel_group_npu
Bosco-lab Nov 27, 2024
2ed1460
Merge pull request #82 from Bosco-lab/rc4main
hust17yixuan Nov 29, 2024
a752d17
modify internal calls of npu boxes_overlap_bev & box_iou_rotated
OlaWod Dec 16, 2024
b5c97d4
Merge pull request #83 from OlaWod/boxes
hust17yixuan Dec 18, 2024
0aa84d2
git checkout origin pixel_group
Annarine Jan 7, 2025
10d0767
Merge pull request #84 from Annarine/rc4main
hust17yixuan Jan 7, 2025
540ac07
add impl of assign_score_withk backward
Hua-yuxiu Jan 7, 2025
4095faf
Revert "modify internal calls of npu boxes_overlap_bev & box_iou_rota…
hust17yixuan Jan 8, 2025
721408e
Merge pull request #86 from Hua-yuxiu/rc4main
hust17yixuan Jan 13, 2025
0937678
add diou_npu
Jan 13, 2025
d45099a
Merge pull request #87 from JYYCaN/rc4main
hust17yixuan Jan 14, 2025
3dda925
Update box_iou_rotated_npu.cpp
JYYCaN Jan 23, 2025
bf2bc16
Merge pull request #88 from JYYCaN/rc4main
hust17yixuan Feb 5, 2025
2cb2a34
fix hccl compile
Feb 6, 2025
27274df
Update boxes_overlap_bev C++ adapt file
JYYCaN Feb 19, 2025
7742a88
Merge pull request #89 from JYYCaN/rc4main
hust17yixuan Feb 19, 2025
358a4ff
Update boxes_overlap_bev_npu.cpp
JYYCaN Feb 25, 2025
058e010
Merge pull request #90 from JYYCaN/patch-3
hust17yixuan Feb 25, 2025
7066523
remove points_in_polygons constraints in text
hust17yixuan Mar 24, 2025
fa45521
fix border_align_npu lint
hust17yixuan Mar 24, 2025
f3eb9c7
fix points_in_box_npu_all lint
hust17yixuan Mar 24, 2025
a5d21aa
fix roi_algin_rotated ops
hust17yixuan Mar 24, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 2 additions & 5 deletions mmcv/ops/csrc/pytorch/npu/box_iou_rotated_npu.cpp
Original file line number Diff line number Diff line change
@@ -11,11 +11,8 @@ void box_iou_rotated_npu(const Tensor boxes1, const Tensor boxes2, Tensor ious,

TORCH_CHECK(boxes1.size(1) == 5, "boxes1 must be 2D tensor (N, 5)");
TORCH_CHECK(boxes1.size(1) == 5, "boxes1 must be 2D tensor (N, 5)");

auto trans = false;
auto is_clockwise = false;
EXEC_NPU_CMD(aclnnBoxesOverlapBev, boxes1, boxes2, trans, is_clockwise,
aligned, mode_flag, ious);

EXEC_NPU_CMD(aclnnBoxIou, boxes1, boxes2, mode_flag, aligned, ious);
return;
}