Skip to content

Pull requests: Lightning-AI/lightning-thunder

New pull request New
47 Open 1,094 Closed
47 Open 1,094 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

[TEST] specify device_id in init_process_group
#1802 opened Feb 25, 2025 by Aidyn-A Loading…
[ThunderFX report]: Improve the interfaces
#1800 opened Feb 25, 2025 by kiya00 Loading…
4 tasks
16
Update cuDNN SDPA execution functions to use the strides of attn_mask cudnn executors strides
#1798 opened Feb 25, 2025 by IvanYashchuk Loading…
1
allow embedding without renorm to be lowered as nvfuser index_select
#1796 opened Feb 24, 2025 by jjsjann123 • Draft
restore nvfuser take_along_axis support
#1795 opened Feb 24, 2025 by jjsjann123 • Draft
Update test_vjp_correctness_sdpa_manual with fp16/bf16 friendly scale value
#1791 opened Feb 24, 2025 by riccardofelluga Loading…
1
5
add torch.optim._functional.rmsprop benchmark
#1788 opened Feb 22, 2025 by khushi-411 Loading…
2 of 4 tasks
1
Fix tensor.to
#1787 opened Feb 21, 2025 by riccardofelluga Loading…
1
1
Split get_computation_and_inputs
#1784 opened Feb 20, 2025 by t-vi • Draft
[WIP] Update aliases in-place operators
#1780 opened Feb 18, 2025 by beverlylytle • Draft
26
benchmark_litgpt - update low_precision_mode accepted values
#1779 opened Feb 18, 2025 by kshitij12345 • Draft
1
Avoid name collisions in _convert_pytorchfunc_to_thundertrace
#1777 opened Feb 18, 2025 by crcrpar • Draft
1
Remove requires_grad specification from meta functions autograd
#1769 opened Feb 17, 2025 by IvanYashchuk Loading…
add torch.optim._functional.sgd benchmark
#1766 opened Feb 15, 2025 by khushi-411 Loading…
2 of 4 tasks
2
add adam optimizer benchmark
#1764 opened Feb 13, 2025 by khushi-411 Loading…
2 of 4 tasks
39
Revert updates to saved_tensors info on fw_extrace, no need to remove duplicate variables autograd
#1756 opened Feb 7, 2025 by IvanYashchuk Loading…
1
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
[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
ProTip! Add no:assignee to see everything that’s not assigned.