Lightning-AI / Lightning-AI/lightning-thunder
test_vjp_correctness_cross_entropy_manual_cross_entropy_nvfuser_cuda_thunder flaky
Open
Nobody has claimed this yet.
ci / tests
- Dominant language
- Python
- Stars
- 1.5k
- Forks
- 121
- PR merge metrics
- No merged PRs in 30d
Description
## 🐛 Bug
thunder/tests/test_grad.py::test_vjp_correctness_cross_entropy_manual_cross_entropy_nvfuser_cuda_thunder::dtypes::float64
is flaky.
### To Reproduce
Shows on CI (with PT main branch)
cc @borda
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.
Research direction
Start with thunder/tests/test_grad.py::test_vjp_correctness_cross_entropy_manual_cross_entropy_nvfuser_cuda_thunder::dtypes::float64 and reproduce it in CI with the PT main branch. Trace the flaky result and verify the test runs reliably for the reported CUDA, nvFuser, Thunder, and float64 combination.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- compilers, testing-qa
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100