Skip to content

Pull requests: Lightning-AI/lightning-thunder

New pull request New
36 Open 976 Closed
36 Open 976 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

TE: avoid proxy name clash TransformerEngine
#1626 opened Jan 9, 2025 by kshitij12345 Loading…
TE: disable recomputing in backward TransformerEngine
#1625 opened Jan 9, 2025 by kshitij12345 Loading…
1
fix no_grad and torch.ops.higher_order.autograd_function_apply interaction
#1620 opened Jan 8, 2025 by kshitij12345 Loading…
1
Add a new dynamo.last_traces entry point.
#1614 opened Jan 7, 2025 by tfogal • Draft
6
jit : update how nn.Parameters are detected and tagged with STATIC_MEMORY_LOCATION tag
#1611 opened Jan 7, 2025 by kshitij12345 Loading…
4
Remove requires_grad from regular TensorProxy and add specialized TensorProxy with requires_grad and is_leaf attributes developer efficiency
#1599 opened Dec 30, 2024 by IvanYashchuk • Draft
5
Allow Proxy creation without a set TraceCtx tracing architecture
#1598 opened Dec 30, 2024 by IvanYashchuk • Draft
6
Make TensorProxy requires_grad argument to be False by default tracing architecture
#1597 opened Dec 30, 2024 by IvanYashchuk • Draft
[tensor subclass] print type_string of tensor attributes
#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…
clean up imports of jit_ext.py
#1588 opened Dec 27, 2024 by crcrpar Loading…
1
[tensor wrapper subclass] Add support for torchao.float8 mlp
#1585 opened Dec 23, 2024 by crcrpar • Draft
[tensor wrapper subclass] Add trace transform for tensor subclasses documentation Improvements or additions to documentation
#1584 opened Dec 23, 2024 by crcrpar • Draft
[tensor wrapper subclass] Add Proxy, prim, and lookaside
#1583 opened Dec 23, 2024 by crcrpar • Draft
[nvfuser executor] Allow sm89 for fp8 types
#1576 opened Dec 20, 2024 by crcrpar • Draft
6
Rename AdHocExecutor -> TransientOperatorExecutor and add documentation
#1573 opened Dec 19, 2024 by IvanYashchuk Loading…
4
update checkpointing support for jit
#1560 opened Dec 17, 2024 by t-vi • Draft
7
ThunderFX: handles the callable input of fx.Node
#1548 opened Dec 12, 2024 by kiya00 • Draft
4 tasks
1
18
more general setattribute
#1541 opened Dec 11, 2024 by t-vi • Draft
[benchmark] add option to enable CompiledAutograd
#1536 opened Dec 10, 2024 by crcrpar • Draft
[litgpt benchmark] enable force_recompute_fp8_weight_in_bwd when torchao.float8 is used with FSDP2 benchmarking
#1528 opened Dec 9, 2024 by crcrpar Loading…
Refactor autocast to AutocastTransform for Improved Composability and Transform Integration
#1516 opened Dec 4, 2024 by rittik9 • Draft
1 of 4 tasks
1
2
Fix TransformerEngine and Activation checkpointing interaction for thunderFX
#1473 opened Nov 26, 2024 by kshitij12345 • Draft
2
1
update ctxs of bsym representing torch.autograd.Funcion
#1462 opened Nov 21, 2024 by crcrpar • Draft
Raise when in place operations occur on leafs requiring grad autograd in-place
#1458 opened Nov 20, 2024 by beverlylytle Loading…
4 tasks
1
46
ProTip! Follow long discussions with comments:>50.