Skip to content

Add experimental MLIR debuginfo writer API#6799

Merged
chunnienc merged 8 commits intomasterfrom
chunnienc/init-tag-tensor-api
Mar 26, 2024
Merged

Add experimental MLIR debuginfo writer API#6799
chunnienc merged 8 commits intomasterfrom
chunnienc/init-tag-tensor-api

Conversation

@chunnienc
Copy link
Copy Markdown
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
Copy Markdown
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
Copy Markdown

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
Copy Markdown
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

Comment thread torch_xla/experimental/xla_mlir_debuginfo.py
Comment thread torch_xla/csrc/runtime/xla_mlir_debuginfo_helper.cc
@lsy323
Copy link
Copy Markdown
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
Copy Markdown
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
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