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

[better_errors] Merge the JaxprDebugInfo and TracingDebugInfo into core.DebugInfo #26262

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

gnecula
Copy link
Collaborator

@gnecula gnecula commented Feb 1, 2025

Previously, we had two almost identical classes: TracingDebugInfo and JaxprDebugInfo. The only difference was that TracingDebugInfo had a thunk to return the result paths, while JaxprDebugInfo had the result paths resolved to a tuple. The separation of these types provided some clarity, but also led to code duplication and required conversions as the debugging info goes from WrappedFun to a Jaxpr and then to WrappedFun again.

Much of this PR is just renaming.

@gnecula gnecula self-assigned this Feb 1, 2025
@gnecula gnecula added the pull ready Ready for copybara import and testing label Feb 1, 2025
@gnecula gnecula force-pushed the debug_info_one branch 4 times, most recently from bdc3d7e to 182cb5a Compare February 2, 2025 03:29
@gnecula gnecula requested a review from dfm February 2, 2025 04:20
…re.DebugInfo

Previously, we had two almost identical classes: `TracingDebugInfo` and
`JaxprDebugInfo`. The only difference was that `TracingDebugInfo` had
a thunk to return the result paths, while `JaxprDebugInfo` had the
result paths resolved to a tuple. The separation of these types
provided some clarity, but also led to code duplication and
required conversions as the debugging info goes from `WrappedFun`
to a `Jaxpr` and then to `WrappedFun` again.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant