XRT branch cherry-pick 07/10 #5293
JackCaoG
merged
24 commits into
xrt
from
JackCaoG/xrt_cherry_pick_07_10
Skip calling as_strided in empty_strided_symint if the input has dyna…
efa627b7
Add XRT nightly builds (#5261)
c91738ab
Add ToString method for both PjrtData and PjrtShardedData (#5265)
5575a450
fix xrt tostring
94dfea01
Update Sharded graph HLO dumping (#5266)
d52993d4
Disable Bazel remote cache for forked PR (#5259)
ab179bca
Suppress debug symbols in OpenXLA code (#5269)
3967d7ba
[SPMD] Sharding n-d tensor on (n+1)-d Mesh (#5268)
bfb85fad
Make TPU detection more robust (#5271)
cba267f4
Clean bazel stuff on distutils clean. (#5274)
a2d99acd
fix conflict
e91ad3ad
Fix the error when export_torch_model is given a non-tensor (#5277)
20d29869
Dsiable test_simple_model_with_different_input_shape since it is curr…
6087f8b2
Always do build_ext in python setup.py develop (#5273)
aa112e01
Remove or improve several hardcoded TPU test conditions (#5272)
dd5709da
Add `runtime.host_index` (#5283)
34e9e571
Make it an error if calling sizes() on a dynamic tensor. (#4998)
9c69d835
Fix the error where mark_step does not materalize tensors on SPMD:0 (…
99452b03
Disable torch._dynamo.config.automatic_dynamic_shapes (#5285)
2402f340
[Traceable Collecive] Hide token for all_gather (#5232)
710cc524
Lower squeeze.dims (#5286)
4eb14e7c
avoid copy proto in PrepareOutputShardingPropagation (#5287)
348faba7
Revert "Suppress debug symbols in OpenXLA code (#5269)"
65afe659
Revert "fix conflict"
3444177f
will-cromar
approved these changes
on 2023-07-11
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub