PR #4014 [DDP] Add a test case for XRT

[DDP] Add a test case for XRT #4014

alanwaketan merged 16 commits into master from alanwaketan/xrt

alanwaketan requested a review from

will-cromar 3 years ago

alanwaketan requested a review from

JackCaoG 3 years ago

JackCaoG added ddp

will-cromar approved these changes on 2022-09-16

JackCaoG commented on 2022-09-16

JackCaoG approved these changes on 2022-09-16

Introduce distributed_util.py

098bc1f2

Make PJRT work on V4

4ba775e7

Remove some imports

848222af

Add XRT testcase

9ecf48f1

Fix test_ddp.py

88461915

Fix distributed_util.py

0c808e3b

Fix nn.Lineaer

63059961

Add to run_tests.sh

45ae8999

Fix linters

d5145418

Don't run ddp test on cpu env

f9181b12

Move distributed_util.py to test/

2ba5e6dd

Attempt to fix CI import error

1171f71f

Attempt to fix the import problem again

a5072fd4

Attempt to fix the import problem again and again

1249b88f

alanwaketan force pushed from 75e170ba to 1249b88f 3 years ago

Fix rebase mistakes

54f90b45

fix linter

c427f213

alanwaketan merged 7d6a7234 into master 3 years ago

alanwaketan deleted the alanwaketan/xrt branch 3 years ago

Reviewers

JackCaoG

will-cromar

Assignees

No one assigned

Labels

ddp

Milestone

No milestone

xla [DDP] Add a test case for XRT #4014 Merged

[DDP] Add a test case for XRT #4014

xla
[DDP] Add a test case for XRT
#4014

Merged