xla
Update tf pin to 03/2023
#4840
Merged

Update tf pin to 03/2023 #4840

JackCaoG merged 25 commits into master from tf-update-032023
wonjoo-wj
wonjoo-wj wonjoo-wj force pushed from 56a14c3d to acc62d58 3 years ago
wonjoo-wj wonjoo-wj force pushed from acc62d58 to bc21d7b8 3 years ago
wonjoo-wj wonjoo-wj force pushed from fc71abe4 to c93cf792 3 years ago
wonjoo-wj wonjoo-wj force pushed from 8d0a5415 to f1e37bda 3 years ago
lsy323 lsy323 force pushed from f1e37bda to 90537035 3 years ago
will-cromar
wonjoo-wj
wonjoo-wj wonjoo-wj force pushed from 7682dd17 to 5b943773 3 years ago
wonjoo-wj wonjoo-wj force pushed from 1e10aac4 to 68a61d1d 3 years ago
wonjoo-wj wonjoo-wj assigned wonjoo-wj wonjoo-wj 2 years ago
wonjoo-wj wonjoo-wj force pushed from fbdad346 to 93919302 2 years ago
wonjoo-wj wonjoo-wj force pushed from 93919302 to 717d9a37 2 years ago
JackCaoG
wonjoo-wj wonjoo-wj force pushed from 717d9a37 to e7a11424 2 years ago
wonjoo-wj wonjoo-wj force pushed from 6618f70b to ba043602 2 years ago
wonjoo-wj wonjoo-wj force pushed from d5aa01e0 to e3965e24 2 years ago
wonjoo-wj Update grpc_version.diff to avoid merge conflicts
f94e1f2d
wonjoo-wj Remove tf/patches_cudnn_int8_32.diff
8588f146
wonjoo-wj Update workspace to fetch new TF pin
aae7ea53
wonjoo-wj Update WORKSPACE for CI testing
167784e3
wonjoo-wj Add local_endezvous.diff and topk_rewriter.diff tf patches
dc9a07b3
wonjoo-wj Update WORKSPACE for testing
9ddc70de
wonjoo-wj Add tf patch for triton filesystem
1c8e6852
wonjoo-wj Update topk_rewriter.diff
c5a5c30e
wonjoo-wj Update WORKSPACE to include triton_filesystem.diff
67179f2a
wonjoo-wj Update triton_filesystem.diff to include linkops
38f7f34a
Add -lstdc++fs to Triton BUILD rules
ecab8655
wonjoo-wj Update libtpu to 03/30
eadbfb82
wonjoo-wj Disable tests for CI testing
a49db1d6
wonjoo-wj Update PJRT use_major_to_minor_data_layout_for_callbacks flag to true
9b6f60db
will-cromar Use `logical_on_device_shape` now that it's in libtpu
504d7646
will-cromar Allocate a literal with bounded shape in TransferFromServer
1cf2d27b
will-cromar formatting
4632b1b6
wonjoo-wj Revert "Disable tests for CI testing"
40e7ecb6
wonjoo-wj Skip test_fsdp_auto_wrap and test_torch_distributed_fsdp_meta.py on GPU
d6c22f43
wonjoo-wj Fix imports for unittests
83f75ba2
wonjoo-wj Skip test_torch_distributed_fsdp_meta.py on GPU
e44652c7
wonjoo-wj Run linter
d1f7df0d
wonjoo-wj Skip test_train_mp_imagenet_fsdp in .circleci/common.sh
f399816b
wonjoo-wj wonjoo-wj force pushed from e3965e24 to f399816b 2 years ago
wonjoo-wj Remove supports_logical_on_device_shape_ from pjrt_computation_client…
1647d7ab
JackCaoG
JackCaoG commented on 2023-04-19
JackCaoG
JackCaoG commented on 2023-04-19
JackCaoG
JackCaoG commented on 2023-04-19
JackCaoG
JackCaoG commented on 2023-04-19
JackCaoG
JackCaoG commented on 2023-04-19
JackCaoG
JackCaoG commented on 2023-04-19
JackCaoG
JackCaoG approved these changes on 2023-04-19
wonjoo-wj
wonjoo-wj Disable test_train_mp_imagenet_fsdp.py tests in .circleci/common.sh
d06fd02c
wonjoo-wj
JackCaoG JackCaoG added tf-update
JackCaoG JackCaoG marked this pull request as ready for review 2 years ago
wonjoo-wj wonjoo-wj changed the title [WIP] Update tf pin to 03/2023 Update tf pin to 03/2023 2 years ago
wonjoo-wj
JackCaoG JackCaoG merged c1138e88 into master 2 years ago
lsy323 lsy323 deleted the tf-update-032023 branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone