Skip to content

Comments

DTensor: NVFuser Integration#2177

Merged
t-vi merged 12 commits intoLightning-AI:mainfrom
kshitij12345:dtensor-nvfuser-integration
Jun 25, 2025
Merged

DTensor: NVFuser Integration#2177
t-vi merged 12 commits intoLightning-AI:mainfrom
kshitij12345:dtensor-nvfuser-integration

Conversation

@kshitij12345
Copy link
Collaborator

@kshitij12345 kshitij12345 commented Jun 3, 2025 •
edited
Loading

Builds on #1907

TODO:

  • Add test for this integration.
  • Make sure this works with PyTorch built with USE_DISTRIBUTED=0

kshitij12345
kshitij12345 commented Jun 3, 2025
View reviewed changes
kshitij12345
kshitij12345 commented Jun 3, 2025
View reviewed changes
@kshitij12345 kshitij12345 changed the title [DO NOT MERGE] WIP DTensor NVFuser Integration WIP DTensor NVFuser Integration Jun 16, 2025
@kshitij12345 kshitij12345 requested a review from wujingyue June 16, 2025 13:54
@kshitij12345 kshitij12345 force-pushed the dtensor-nvfuser-integration branch from 9900b78 to 512dc1a Compare June 16, 2025 14:07
@github-actions github-actions bot added documentation Improvements or additions to documentation install docker ci labels Jun 16, 2025
@github-actions github-actions bot removed documentation Improvements or additions to documentation install docker ci labels Jun 16, 2025
@kshitij12345 kshitij12345 changed the title WIP DTensor NVFuser Integration DTensor: NVFuser Integration Jun 16, 2025
IvanYashchuk
IvanYashchuk reviewed Jun 17, 2025
View reviewed changes
wujingyue
wujingyue reviewed Jun 17, 2025
View reviewed changes
IvanYashchuk
IvanYashchuk approved these changes Jun 18, 2025
View reviewed changes
@kshitij12345 kshitij12345 marked this pull request as ready for review June 18, 2025 10:53
@kshitij12345 kshitij12345 added nvfuser DTensor Issues about DTensor support in Thunder labels Jun 18, 2025
@kshitij12345
Copy link
Collaborator Author

kshitij12345 commented Jun 18, 2025

Ping @t-vi for stamp

wujingyue
wujingyue approved these changes Jun 18, 2025
View reviewed changes
crcrpar
crcrpar reviewed Jun 20, 2025
View reviewed changes
@kshitij12345
Copy link
Collaborator Author

kshitij12345 commented Jun 25, 2025

Ping @t-vi for stamp

t-vi
t-vi approved these changes Jun 25, 2025
View reviewed changes
Copy link
Collaborator

@t-vi t-vi left a comment •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you @kshitij12345 @wujingyue @IvanYashchuk @crcrpar

@t-vi t-vi enabled auto-merge (squash) June 25, 2025 14:58
@t-vi t-vi merged commit 898407b into Lightning-AI:main Jun 25, 2025
49 checks passed
@wujingyue wujingyue mentioned this pull request Jun 30, 2025
Merged
@github-actions github-actions bot deleted the dtensor-nvfuser-integration branch September 20, 2025 00:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@wujingyue wujingyue wujingyue approved these changes

@crcrpar crcrpar crcrpar left review comments

@IvanYashchuk IvanYashchuk IvanYashchuk approved these changes

@t-vi t-vi t-vi approved these changes

@mruberry mruberry Awaiting requested review from mruberry mruberry is a code owner

@lantiga lantiga Awaiting requested review from lantiga lantiga is a code owner

Assignees

No one assigned

Labels

DTensor Issues about DTensor support in Thunder nvfuser

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

5 participants

@kshitij12345 @wujingyue @crcrpar @IvanYashchuk @t-vi