PR #2966 Support skip_first_batches for XLA

Support skip_first_batches for XLA #2966

muellerzr merged 3 commits into huggingface:main from AlibabaPAI:fix_xla_skip_data

Fix skip_first_batches for XLA

1d2dc00b

muellerzr commented on 2024-07-29

Use state to check XLA

5688e5e7

Change to PartialState

f468964d

muellerzr approved these changes on 2024-08-01

muellerzr merged 79ca85c2 into main 1 year ago

yitongh deleted the fix_xla_skip_data branch 1 year ago

Reviewers

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone