Skip to content

We Need Compile Support For Mamba! #31246

@zhenglongjiepheonix

Description

@zhenglongjiepheonix

Feature request

This feature adds torch.compile support for mamba archtecture

Motivation

The motivation is that by supporting compile on mamba, we can get faster inference speed and better throughput even if we don't have high performance specified mamba kernels installed!

Your contribution

I will be implementing this feature

Metadata

Metadata

Labels

CompilationIssues related to torchdynamo and torchinductorFeature requestRequest for a new feature

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions