PR #2263 Fix breakpoint API in test_script.py on TPU.

Fix breakpoint API in test_script.py on TPU. #2263

muellerzr merged 5 commits into huggingface:main from vanbasten23:fix_check_trigger

Fix breakpoint API in test_script.py on TPU.

66451896

muellerzr requested changes on 2023-12-15

only call set_trigger on the main process

0be0da28

The test passed.

4be48eba

add a comment

b9a4308c

vanbasten23 requested a review from

muellerzr 2 years ago

will-cromar commented on 2024-01-02

Call mark_step after all_reduce to make torch_xla run collective op l…

4fab8000

vanbasten23 requested a review from

will-cromar 2 years ago

will-cromar approved these changes on 2024-01-04

muellerzr approved these changes on 2024-01-05

muellerzr merged 2cfa88bd into main 2 years ago

Reviewers

muellerzr

will-cromar

Assignees

No one assigned

Labels

None yet

Milestone

No milestone