yaoshiang comments

Results 15 comments of


                                            yaoshiang

Document on how to profile with torch_xla

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...

Move env set-up from test/run_tests.sh to individual .py files

This would also enable integration with VSCode's visual python debugging of unit tests.

Migrate `torch_xla.device()` to `torch.device('xla')`

Thanks for this good analysis. Clearly the APIs are not the same, but, do they have enough expressiveness to do the same thing? setting an ordinal for device number appears...

Migrate `torch_xla.device()` to `torch.device('xla')`

Thanks. I don't know what is the setter for `torch_xla._XLAC._xla_get_default_device()`, but the fact that is in _XLAC but not exposed properly outside of it is already a redflag this part...

Pypi says PT/XLA 2.6.1 supports any python >= 3.8. But PT/XLA not listed in pip for python 3.12

![Image](https://github.com/user-attachments/assets/6f7033df-93c8-41f6-8697-e45650f0e7b7)