Skip to content

[Feature Request] Support Ascend NPU (CANN) for Mamba3 #933

@BaiH-ang

Description

@BaiH-ang

I would like to see this model gain support for running on Ascend NPUs via the CANN software stack. As PyTorch already supports the Ascend backend through torch_npu, the porting process might be achievable by adapting the CUDA-specific code.

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