pytorch/xla

Pull Requests Commits

Revert "fix conflict"

JackCaoG committed 2 years ago

3444177f

Revert "Suppress debug symbols in OpenXLA code (#5269)"

JackCaoG committed 2 years ago

65afe659

avoid copy proto in PrepareOutputShardingPropagation (#5287)

baoleai committed 2 years ago

348faba7

Lower squeeze.dims (#5286)

wonjoo-wj committed 2 years ago

4eb14e7c

[Traceable Collecive] Hide token for all_gather (#5232)

alanwaketan committed 2 years ago

710cc524

Disable torch._dynamo.config.automatic_dynamic_shapes (#5285)

wonjoo-wj committed 2 years ago

2402f340

Fix the error where mark_step does not materalize tensors on SPMD:0 (#5281)

JackCaoG committed 2 years ago

99452b03

Make it an error if calling sizes() on a dynamic tensor. (#4998)

vanbasten23 committed 2 years ago

9c69d835

Add `runtime.host_index` (#5283)

will-cromar committed 2 years ago

34e9e571

Remove or improve several hardcoded TPU test conditions (#5272)

will-cromar committed 2 years ago

dd5709da

Always do build_ext in python setup.py develop (#5273)

qihqi committed 2 years ago

aa112e01

Dsiable test_simple_model_with_different_input_shape since it is curretnly broken by pytorch (#5282)

JackCaoG committed 2 years ago

6087f8b2

Fix the error when export_torch_model is given a non-tensor (#5277)

qihqi committed 2 years ago

20d29869

JackCaoG committed 2 years ago

e91ad3ad

Clean bazel stuff on distutils clean. (#5274)

stgpetrovic committed 2 years ago

a2d99acd

Make TPU detection more robust (#5271)

will-cromar committed 2 years ago

cba267f4

[SPMD] Sharding n-d tensor on (n+1)-d Mesh (#5268)

khatwanimohit committed 2 years ago

bfb85fad

Suppress debug symbols in OpenXLA code (#5269)

stgpetrovic committed 2 years ago

3967d7ba

Disable Bazel remote cache for forked PR (#5259)

lsy323 committed 2 years ago

ab179bca

Update Sharded graph HLO dumping (#5266)

JackCaoG committed 2 years ago

d52993d4

fix xrt tostring

JackCaoG committed 2 years ago

94dfea01

Add ToString method for both PjrtData and PjrtShardedData (#5265)

JackCaoG committed 2 years ago

5575a450

Add XRT nightly builds (#5261)

will-cromar committed 2 years ago

c91738ab

Skip calling as_strided in empty_strided_symint if the input has dynamic dimensions. (#5239)

vanbasten23 committed 2 years ago

efa627b7

Jack cao g/fix spmd buff is null (#5256)

JackCaoG committed 2 years ago

3a8db635

Fix the linter command in the CI (#5254)

vanbasten23 committed 2 years ago

42db7838

Add nightly build for cuda 12 (#5253)

wonjoo-wj committed 2 years ago

4cedb97a

[Functionalization] Remove view in view_symint (#5231)

alanwaketan committed 2 years ago

Verified 1605972c

Add IAM roles for cloudbuild_editors (#5251)

mateuszlewko committed 2 years ago

Verified 04e892fb

Install `expecttest` in xla_test_job.yaml (#5252)

will-cromar committed 2 years ago

Verified ebd95a65

Older