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 1 year ago

ghpvnist force pushed from b9ab7510 to 55bd4557 1 year ago

ghpvnist force pushed from 7b082fee to 9c34518a 1 year ago

ghpvnist marked this pull request as ready for review 1 year ago

ghpvnist requested a review from

tengyifei 1 year ago

tengyifei requested changes on 2025-05-06

tengyifei requested changes on 2025-05-07

ghpvnist force pushed from f1c556f6 to 06eff815 1 year ago

ghpvnist force pushed from 06eff815 to a2cb810a 1 year ago

tengyifei approved these changes on 2025-05-08

qihqi commented on 2025-05-09

ghpvnist force pushed from e639de8e to b0f60623 1 year 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 1 year ago

Add documentation and add back deprecation warning

6d576f41

ghpvnist force pushed from 1c554614 to 6d576f41 1 year ago

Fix rebase conflicts

1e5883d1

ghpvnist requested a review from

qihqi 1 year ago

ghpvnist assigned

ghpvnist 1 year ago

Fix tests

1607728d

qihqi approved these changes on 2025-05-15

ghpvnist merged bbafe357 into master 363 days ago

ghpvnist deleted the sync branch 363 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