-
Notifications
You must be signed in to change notification settings - Fork 31.2k
Closed
Labels
CompilationIssues related to torchdynamo and torchinductorIssues related to torchdynamo and torchinductorFeature requestRequest for a new featureRequest for a new feature
Description
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
Assignees
Labels
CompilationIssues related to torchdynamo and torchinductorIssues related to torchdynamo and torchinductorFeature requestRequest for a new featureRequest for a new feature