Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Trace Transformation Tutorial - CPU Offloading #966

Merged
merged 15 commits into from
Aug 21, 2024
Merged

Trace Transformation Tutorial - CPU Offloading #966

Show file tree
Hide file tree
Changes from 11 commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
3d602d1
Trace Transformation Tutorial - CPU Offloading
kshitij12345 Aug 13, 2024
d0303d6
don't run larger model on CI
kshitij12345 Aug 13, 2024
689636f
update
kshitij12345 Aug 14, 2024
04a1b54
update
kshitij12345 Aug 14, 2024
f9242f4
Merge branch 'main' of github.com:Lightning-AI/lightning-thunder into…
kshitij12345 Aug 15, 2024
80597ac
update
kshitij12345 Aug 15, 2024
da08033
remove giant traces
kshitij12345 Aug 15, 2024
266e4ea
rename executor
kshitij12345 Aug 15, 2024
edc80b4
update docs
kshitij12345 Aug 15, 2024
267bbab
address review
kshitij12345 Aug 16, 2024
ab751ac
address review: add comment
kshitij12345 Aug 16, 2024
86b0c05
address review: part 1
kshitij12345 Aug 16, 2024
091aa9b
use from_bsym_swap_proxies
kshitij12345 Aug 16, 2024
72ca3d9
address review
kshitij12345 Aug 21, 2024
f1dadc4
address review - create new trace instead of modifying old one directly
kshitij12345 Aug 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/source/index.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,7 @@ The compiled function ``jitted_foo`` takes and returns PyTorch tensors, just lik
What's next <intermediate/whats_next>
FSDP Under the Hood Tutorial <notebooks/dev_tutorials/fsdp_tutorial>
Benchmarking Thunder <intermediate/benchmarking>
Writing a Transform <notebooks/writing_a_trace_transform_cpu_offloading>

.. toctree::
:maxdepth: 1
Expand Down
Loading
Loading