[Bug] roi_align_rotated_npu算子问题

Prerequisite

[X] I have searched Issues and Discussions but cannot get the expected help.
[X] The bug has not been fixed in the latest version(https://github.com/open-mmlab/mmcv).

Environment

python3.9 torch2.1.0 torch2.1.0post3 mmcv2.2.0 CANN8.0.RC2.alpha002

Reproduces the problem - code sample

使用mmrotate官方orient_rcnn中的RotatedSingleRoIExtractor报错

Reproduces the problem - command or script

Reproduces the problem - error message

EZ9999: Inner Error! EZ9999: 2024-06-16-16:07:36.000.344 Aicore kernel execute failed, device_id=0, stream_id=2, report_stream_id=2, task_id=4315, flip_num=0, fault kernel_name=te_roialignrotatedgrad_056c4a3260bed2722ace3b6d0c82dae9725df655db2a6287319a3a6908323738_16bca89b54a5f9eddd1714424cd8c8550afd6cfa35fa0ab5bc94891f9abe451c3, fault kernel info ext=te_roialignrotatedgrad_056c4a3260bed2722ace3b6d0c82dae9725df655db2a6287319a3a6908323738_1__kernel0, program id=146, hash=725125899295747968.[FUNC:GetError][FILE:stream.cc][LINE:1502] TraceBack (most recent call last): [AIC_INFO] after execute:args print end[FUNC:GetError][FILE:stream.cc][LINE:1502] rtStreamSynchronize execute failed, reason=[vector core exception][FUNC:FuncErrorReason][FILE:error_message_manage.cc][LINE:53] synchronize stream failed, runtime result = 507035[FUNC:ReportCallError][FILE:log_inner.cpp][LINE:161]

Additional information

在哪个特定的环境下可以跑通roialignrotatedNPU的前向与后向

open-mmlab / mmcv