Skip to content

[Bug] mmcv ops cannot use AICore on torch 2.6.0 on Ascend npu #3324

Description

@Appgye

Prerequisite

Environment

python 3.9
mmcv latest (2.2.0)
torch 2.6.0
CANN 8.5

Reproduces the problem - code sample

install and run under Official Documentation and Examples

Reproduces the problem - command or script

install and run under Official Documentation and Examples

Reproduces the problem - error message

我是从源码进行安装的,安装后运行官方的实例程序可以运行,但是我观察到npu是有显存占用的,但是AICORE一直为0,应该全是cpu进行的计算

Additional information

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions