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

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions