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.

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

Add a new dynamo.last_traces entry point. #1614

Merged
merged 14 commits into from
Jan 21, 2025
Merged

Add a new dynamo.last_traces entry point. #1614

merged 14 commits into from
Jan 21, 2025

Conversation

tfogal
Copy link
Collaborator

@tfogal tfogal commented Jan 7, 2025

I am constantly rewriting this so I thought upstreaming something made sense.

Not married to the implementation, but @t-vi @IvanYashchuk curious for your thoughts on having an explicitly entry point for this?

What does this PR do?

It adds a new entry point which iterates over all the ThunderFX subgraphs and grabs the trace information.

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

thunder/dynamo/utils.py Outdated Show resolved Hide resolved
@tfogal tfogal force-pushed the tfogal/tfx-last-traces branch 3 times, most recently from 030a991 to fcedf0c Compare January 8, 2025 19:03
@tfogal tfogal force-pushed the tfogal/tfx-last-traces branch from 444e30f to 4182e96 Compare January 10, 2025 17:25
@tfogal tfogal marked this pull request as ready for review January 10, 2025 19:38
@t-vi
Copy link
Collaborator

t-vi commented Jan 15, 2025

@IvanYashchuk do you want to take a look before merging?

Copy link
Collaborator

@IvanYashchuk IvanYashchuk left a comment

Choose a reason for hiding this comment

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

Let's merge after @kshitij12345 approves it.

thunder/dynamo/__init__.py Outdated Show resolved Hide resolved
thunder/dynamo/utils.py Outdated Show resolved Hide resolved
thunder/tests/test_dynamo.py Outdated Show resolved Hide resolved
Copy link
Collaborator

@kshitij12345 kshitij12345 left a comment

Choose a reason for hiding this comment

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

Overall LGTM, just a couple of comments and should clean-up the changes pointed by @IvanYashchuk

Thank you @tfogal

thunder/dynamo/compiler.py Show resolved Hide resolved
thunder/dynamo/compiler.py Show resolved Hide resolved
thunder/dynamo/compiler.py Outdated Show resolved Hide resolved
@tfogal tfogal force-pushed the tfogal/tfx-last-traces branch from 4182e96 to 809490e Compare January 15, 2025 19:03
Copy link
Collaborator

@t-vi t-vi left a comment

Choose a reason for hiding this comment

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

Stamping based on @kshitij12345 's review. Thank you @tfogal @kshitij12345

@t-vi t-vi enabled auto-merge (squash) January 21, 2025 11:01
@t-vi t-vi merged commit 531969a into main Jan 21, 2025
49 checks passed
@t-vi t-vi deleted the tfogal/tfx-last-traces branch January 21, 2025 11:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

6 participants