Skip to content

Pull requests: Lightning-AI/lightning-thunder

New pull request New
41 Open 1,068 Closed
41 Open 1,068 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Revert "fuse only consecutive symbols (#1608)" fusion logic
#1763 opened Feb 11, 2025 by IvanYashchuk • Draft
Remove unused but annotated out from OpExProcessor.process_bsym
#1761 opened Feb 11, 2025 by crcrpar Loading…
Adds the timer based on torch.profiler.profile
#1759 opened Feb 10, 2025 by kiya00 • Draft
4 tasks
1
26
Revert updates to saved_tensors info on fw_extrace, no need to remove duplicate variables autograd
#1756 opened Feb 7, 2025 by IvanYashchuk Loading…
Update vjp implementation to attach residuals to all outputs autograd
#1755 opened Feb 7, 2025 by IvanYashchuk • Draft
2
Simplify and fix match_fw_and_bw_saved_for_bw_proxies implementation tracing architecture
#1754 opened Feb 7, 2025 by IvanYashchuk • Draft
2
fixing out of bound access for nll_loss
#1752 opened Feb 7, 2025 by jjsjann123 Loading…
1
3
[WIP - CI test] Perform rms_norm in fp32
#1751 opened Feb 6, 2025 by riccardofelluga • Draft
[nvfuserex] Decomposed torch._scaled_mm
#1749 opened Feb 6, 2025 by crcrpar • Draft
numpy 2.0 dependencies
#1748 opened Feb 6, 2025 by t-vi Loading…
Adds a benchmark of MoE layer with deepseek-ai/DeepSeek-R1 config benchmarking
#1746 opened Feb 5, 2025 by kiya00 Loading…
4 tasks
16
Add benchmark isolation option benchmarking documentation Improvements or additions to documentation
#1719 opened Jan 30, 2025 by riccardofelluga Loading…
Relax test_te_with_retain_graph xfail strictness in test_transformer_engine_executor.py
#1718 opened Jan 30, 2025 by nWEIdia Loading…
3 tasks
FIx typos in adding_custom_operator.ipynb
#1717 opened Jan 29, 2025 by qqaatw Loading…
4 tasks
1
Grad transform patch
#1712 opened Jan 29, 2025 by jjsjann123 • Draft
10
[DO NOT REVIEW][SMOKE TEST] Skip prims shape in grad transform
#1704 opened Jan 27, 2025 by jjsjann123 • Draft
5
[WIP] Test out thunder.jit w/ NeMo models.
#1694 opened Jan 24, 2025 by tfogal • Draft
3
bump litgpt and bitsandbytes dependencies
#1691 opened Jan 24, 2025 by t-vi • Draft
Remove prims.embedding and prims.embedding_backward operators
#1689 opened Jan 24, 2025 by IvanYashchuk Loading…
10
make no_sync behave like in pytorch
#1681 opened Jan 22, 2025 by t-vi • Draft
1
[WIP] initial changes to accommodate pipeline parallelism transform
#1650 opened Jan 16, 2025 by ali-alshaar7 • Draft
4 tasks
revise examine memory's collection handling, output memory annotation in benchmark_litgpt
#1634 opened Jan 10, 2025 by t-vi • Draft
Remove requires_grad from regular TensorProxy and add specialized TensorProxy with requires_grad and is_leaf attributes ci dependencies developer efficiency documentation Improvements or additions to documentation install
#1599 opened Dec 30, 2024 by IvanYashchuk • Draft
8
[tensor subclass] print type_string of tensor attributes documentation Improvements or additions to documentation
#1592 opened Dec 29, 2024 by crcrpar Loading…
1
check tensor attrs of tensor wrapper subclasses in prologue
#1591 opened Dec 28, 2024 by crcrpar Loading…
ProTip! Updated in the last three days: updated:>2025-02-08.