xla icon indicating copy to clipboard operation
xla copied to clipboard

Document on how to profile with torch_xla

Open lsy323 opened this issue 9 months ago • 6 comments

📚 Documentation

I found we don't have a doc/guide on how to profile with torch_xla. We should add this because getting profile is essential for performance analysis.

lsy323 avatar Feb 26 '25 23:02 lsy323

@mikegre-google is this something you can help with?

miladm avatar Feb 28 '25 18:02 miladm

Yes, absolutely. Who can I work with to gather the information needed?

mikegre-google avatar Feb 28 '25 19:02 mikegre-google

I have not heard back from anyone on this. I'll need someone to contact me to get me the information you would like added to the documentation. Thanks!

mikegre-google avatar May 30 '25 15:05 mikegre-google

There is a profiling guide hidden as a section of the "intro", here.

https://github.com/pytorch/xla/blob/master/docs/source/learn/xla-overview.md

It was my intention to break this out into our own profiling guide, however, there is another profiling guide on cloud.google.com, so there is more work to think through this properly get it right with xprof, etc. @chrisjones is now leading all this. (I don't have his github handle).

yaoshiang avatar May 30 '25 17:05 yaoshiang

Reached out to @chrisjones, will update when I have more information. In the meantime are there torch/xla specific profiling topics we want to address?

mikegre-google avatar Jun 03 '25 18:06 mikegre-google

Will meet with @chrisjones early next week to discuss.

mikegre-google avatar Jun 03 '25 18:06 mikegre-google