Skip to content

Pull requests: Lightning-AI/lightning-thunder

New pull request New
39 Open 1,073 Closed
39 Open 1,073 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

add torch.optim._functiona.sgd benchmark
#1766 opened Feb 15, 2025 by khushi-411 Loading…
2 of 4 tasks
Add fusion_type compile option for controlling the fusion logic fusion logic
#1765 opened Feb 14, 2025 by IvanYashchuk • Draft
2
add adam optimizer benchmark
#1764 opened Feb 13, 2025 by khushi-411 Loading…
2 of 4 tasks
8
Revert "fuse only consecutive symbols (#1608)" fusion logic
#1763 opened Feb 11, 2025 by IvanYashchuk Loading…
5
Remove unused but annotated out from OpExProcessor.process_bsym
#1761 opened Feb 11, 2025 by crcrpar Loading…
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
Relax test_te_with_retain_graph xfail strictness in test_transformer_engine_executor.py
#1718 opened Jan 30, 2025 by nWEIdia Loading…
3 tasks
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…
[tensor wrapper subclass] Add support for torchao.float8 mlp
#1585 opened Dec 23, 2024 by crcrpar • Draft
1
[tensor wrapper subclass] Add trace transform for tensor subclasses documentation Improvements or additions to documentation
#1584 opened Dec 23, 2024 by crcrpar • Draft
ProTip! What’s not been updated in a month: updated:<2025-01-15.