Lightning-AI / Lightning-AI/lightning-thunder
[Pattern matching] Replace computation patterns on the FX Graph level
Open
@riccardofelluga is already working on this.
Since Nov 18, 2025.
pattern matching
- Dominant language
- Python
- Stars
- 1.5k
- Forks
- 121
- PR merge metrics
- No merged PRs in 30d
Description
## 🚀 Feature
This is a sub-issue for the pattern matching feature. To mark this done, the idea is to be able to trace and replace a known pattern from #2693.
This work includes figuring out how `torch.fx.replace_pattern` work and what is needed to use it successfully. Moreover, it includes creating a branch with the needed changes in the `ThunderCompiler`.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.