Lightning-AI / Lightning-AI/lightning-thunder
Test dtype consistency against torch
Open
@IvanYashchuk is already working on this.
Since Apr 30, 2025.
enhancement
operators
testing
- Dominant language
- Python
- Stars
- 1.5k
- Forks
- 121
- PR merge metrics
- No merged PRs in 30d
Description
🚀 Feature
From the conversation in #1970 it would be good to have tests that check the correct output dtype but also that verify that the meta dtype is the same as the output.
In my mind this applies mostly to binary elementwise ops than anything else, but suggestions are very welcome
cc @mruberry
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.