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 experimental MLIR debuginfo writer API #6799

Merged
merged 8 commits into from
Mar 26, 2024

Conversation

chunnienc
Copy link
Collaborator

Experimental API to associate a string with a XLA tensor, which will eventually be propagated to the MLIR NameLoc in the lowerings.

Internal reference: https://b.corp.google.com/issues/330605426

@chunnienc chunnienc requested a review from qihqi March 21, 2024 17:35
@chunnienc
Copy link
Collaborator Author

@qihqi I don't know what max_call_stack_depth means in _set_xla_custom_op_name_prefix and I set it to an arbitrary value 0. Please let me know if there is a more meaningful setting here, thanks.

@lina128
Copy link

lina128 commented Mar 21, 2024

Looking forward to testing this change! Thank you!

@chunnienc chunnienc requested a review from lsy323 March 25, 2024 17:33
Copy link
Collaborator

@lsy323 lsy323 left a comment

Choose a reason for hiding this comment

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

Left some nit and questions. Overall LGTM, thanks @chunnienc

@lsy323
Copy link
Collaborator

lsy323 commented Mar 26, 2024

@chunnienc Could you rebase to torch_xla HEAD? Seems there was a PyTorch PR reverted causing the torch_xla HEAD broken earlier today. The corresponding PR (#6819) is reverted in PyTorch/XLA as well. So rebase to HEAD should resolve the CI failure.

@lsy323
Copy link
Collaborator

lsy323 commented Mar 26, 2024

@chunnienc Looks like your local master is not updated? Saw some unexpected changes in the diff

@chunnienc chunnienc merged commit 64a146d into master Mar 26, 2024
17 of 20 checks passed
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.

3 participants