xla
c1138e88 - Update tf pin to 03/2023 (#4840)

Commit
2 years ago
Update tf pin to 03/2023 (#4840) * Update grpc_version.diff to avoid merge conflicts * Remove tf/patches_cudnn_int8_32.diff * Update workspace to fetch new TF pin * Update WORKSPACE for CI testing * Add local_endezvous.diff and topk_rewriter.diff tf patches * Update WORKSPACE for testing * Add tf patch for triton filesystem * Update topk_rewriter.diff * Update WORKSPACE to include triton_filesystem.diff * Update triton_filesystem.diff to include linkops * Add -lstdc++fs to Triton BUILD rules * Update libtpu to 03/30 * Disable tests for CI testing * Update PJRT use_major_to_minor_data_layout_for_callbacks flag to true * Use `logical_on_device_shape` now that it's in libtpu * Allocate a literal with bounded shape in TransferFromServer * formatting * Revert "Disable tests for CI testing" This reverts commit 68a61d1d79bdac3cc7935ccb5f0e33c5345b2987. * Skip test_fsdp_auto_wrap and test_torch_distributed_fsdp_meta.py on GPU * Fix imports for unittests * Skip test_torch_distributed_fsdp_meta.py on GPU * Run linter * Skip test_train_mp_imagenet_fsdp in .circleci/common.sh * Remove supports_logical_on_device_shape_ from pjrt_computation_client.cc after rebasing from master * Disable test_train_mp_imagenet_fsdp.py tests in .circleci/common.sh --------- Co-authored-by: CircleCI <circleci.ossci@gmail.com> Co-authored-by: Will Cromar <wcromar@google.com>
Author
Parents
Loading