Skip to content

Pull requests: Lightning-AI/lightning-thunder

New pull request New
35 Open 1,028 Closed
35 Open 1,028 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

patching shape query in TensorProxy to avoid prims
#1696 opened Jan 25, 2025 by jjsjann123 • Draft
[WIP] Test out thunder.jit w/ NeMo models.
#1694 opened Jan 24, 2025 by tfogal • Draft
2
Backward transform dependency fix
#1693 opened Jan 24, 2025 by jjsjann123 • Draft
fix: handle buffer setattr at root module
#1692 opened Jan 24, 2025 by ali-alshaar7 Loading…
1 of 4 tasks
bump litgpt and bitsandbytes dependencies
#1691 opened Jan 24, 2025 by t-vi Loading…
fix TE distributed tests with fsdp_v2 and ddp_v2
#1690 opened Jan 24, 2025 by kshitij12345 Loading…
1
Remove prims.embedding and prims.embedding_backward operators
#1689 opened Jan 24, 2025 by IvanYashchuk • Draft
2 tasks
4
[WIP] Add utility function to benchmark performance of fusion region with nvfuser and torch.compile
#1682 opened Jan 22, 2025 by kshitij12345 Loading…
13
make no_sync behave like in pytorch
#1681 opened Jan 22, 2025 by t-vi • Draft
1
add debug option to show interpreter progress
#1680 opened Jan 22, 2025 by t-vi Loading…
fixing dce pass to remove duplicate shape queries.
#1673 opened Jan 21, 2025 by jjsjann123 Loading…
2
[WIP] Add cross_entropy to torchcompile_cat executor
#1655 opened Jan 17, 2025 by riccardofelluga • Draft
1
2
[WIP] initial changes to accommodate pipeline parallelism transform
#1650 opened Jan 16, 2025 by ali-alshaar7 • Draft
4 tasks
[WIP] Thunderfx report
#1636 opened Jan 10, 2025 by kiya00 • Draft
4 tasks
9
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 developer efficiency
#1599 opened Dec 30, 2024 by IvanYashchuk • Draft
7
Allow Proxy creation without a set TraceCtx tracing architecture
#1598 opened Dec 30, 2024 by IvanYashchuk • Draft
6
[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…
[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
[tensor wrapper subclass] Add Proxy, prim, and lookaside
#1583 opened Dec 23, 2024 by crcrpar • Draft
12
[nvfuser executor] Allow sm89 for fp8 types
#1576 opened Dec 20, 2024 by crcrpar • Draft
6
[benchmark] add option to enable CompiledAutograd
#1536 opened Dec 10, 2024 by crcrpar • Draft
Refactor autocast to AutocastTransform for Improved Composability and Transform Integration
#1516 opened Dec 4, 2024 by rittik9 • Draft
1 of 4 tasks
1
2
ProTip! no:milestone will show everything without a milestone.