OpenXLA pin update 07052023 #5304
JackCaoG
merged
28 commits into
master
from
lsiyuan/openxla-0705-pin
vanbasten23
commented
on 2023-07-18
move openxla pin, remove patches
3cb30fb7
Update PartialTile API
46a05128
fake fix for GetStreamExecutorGpuClient
97200d9a
remove topk, se patch
c1c8b5aa
update libtpu version
683f3ebd
fix sharding test
ceb17d34
fix GetStreamExecutorGpuClient
e7e014d6
enhancement on GetStreamExecutorGpuClient fix
68e7ac81
add patch to fix tpu init failure
83513fc4
remove cuda patch
7114102e
fix build with gcc11
d3c4a03a
fix stablehlo dump test error
94e47e83
add constexpr patch
3b2d8d84
fix GetStreamExecutorGpuClient
1ea7f07a
fix linter
bc403563
remove pjrt c client patch
5398a3a0
use pjrt local num process for num_nodes
384d0303
skip test_stack_pred
398209d8
skip ddp tests
215ba40c
disable test_mp_sync_batch_norm
db67be8c
try to fix nccl shmopen error by increasing shm size to 16g
cc8a407a
re-enable ddp tests
f3505c88
try disable large net in pjrt ddp test
093bd49f
Revert "try disable large net in pjrt ddp test"
3fb3fa61
fix shm flag
f7e54ef7
remove env print in CI script
19cc221f
re-enable mp_sync_batch_norm test and refactor test script
7a0bfde2
fix linter, enhance comments
f2aa14fa
JackCaoG
commented
on 2023-07-25
JackCaoG
commented
on 2023-07-25
JackCaoG
approved these changes
on 2023-07-25
will-cromar
approved these changes
on 2023-07-25
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub