xla
Replace implementation of `xla_torch.sync()` to `xm.mark_step()`
#9086

Merged

Replace implementation of `xla_torch.sync()` to `xm.mark_step()` #9086

ghpvnist merged 9 commits into pytorch:master from ghpvnist:sync

ghpvnist force pushed from 8a863988 to b9ab7510 305 days ago

ghpvnist force pushed from b9ab7510 to 55bd4557 305 days ago

ghpvnist force pushed from 7b082fee to 9c34518a 305 days ago

ghpvnist marked this pull request as ready for review 305 days ago

ghpvnist requested a review from

tengyifei 305 days ago

tengyifei requested changes on 2025-05-06

tengyifei requested changes on 2025-05-07

ghpvnist force pushed from f1c556f6 to 06eff815 302 days ago

ghpvnist force pushed from 06eff815 to a2cb810a 302 days ago

tengyifei approved these changes on 2025-05-08

qihqi commented on 2025-05-09

ghpvnist force pushed from e639de8e to b0f60623 297 days ago

Replace implementation of `xla_torch.sync()` to `xm.mark_step()`

93e44b60

Add comment to keep implementation in sync

1ca6403e

Remove change detecting asserts

c68a1a5a

Update debug_util to update index when calling sync

2ff01d0b

Add deprecation warning

293a2dda

Use sync instead of mark_step for test_pt_xla_debug.py

520f7723

ghpvnist force pushed from b0f60623 to 1c554614 297 days ago

Add documentation and add back deprecation warning

6d576f41

ghpvnist force pushed from 1c554614 to 6d576f41 297 days ago

Fix rebase conflicts

1e5883d1

ghpvnist requested a review from

qihqi 297 days ago

ghpvnist assigned

ghpvnist 297 days ago

Fix tests

1607728d

qihqi approved these changes on 2025-05-15

ghpvnist merged bbafe357 into master 295 days ago

ghpvnist deleted the sync branch 295 days ago

Reviewers

qihqi

tengyifei

Assignees

ghpvnist

Labels

None yet

Milestone

No milestone

xla Replace implementation of `xla_torch.sync()` to `xm.mark_step()` #9086 Merged

Replace implementation of `xla_torch.sync()` to `xm.mark_step()` #9086

xla
Replace implementation of `xla_torch.sync()` to `xm.mark_step()`
#9086

Merged