xla
8ffbd1a4 - OpenXLA pin update 07052023 (#5304)

Commit
2 years ago
OpenXLA pin update 07052023 (#5304) * move openxla pin, remove patches * Update PartialTile API * fake fix for GetStreamExecutorGpuClient * remove topk, se patch * update libtpu version * fix sharding test * fix GetStreamExecutorGpuClient * enhancement on GetStreamExecutorGpuClient fix * add patch to fix tpu init failure * remove cuda patch * fix build with gcc11 * fix stablehlo dump test error * add constexpr patch * fix GetStreamExecutorGpuClient * fix linter * remove pjrt c client patch * use pjrt local num process for num_nodes * skip test_stack_pred * skip ddp tests * disable test_mp_sync_batch_norm * try to fix nccl shmopen error by increasing shm size to 16g * re-enable ddp tests * try disable large net in pjrt ddp test * Revert "try disable large net in pjrt ddp test" This reverts commit 093bd49f17b18ff054c5ee592a5488cfd895216c. * fix shm flag * remove env print in CI script * re-enable mp_sync_batch_norm test and refactor test script * fix linter, enhance comments
Author
Parents
Loading