pytorch/xla

Pull Requests Commits

[Functionalization] Remove View in tensor_methods::permute (#5195)

wonjoo-wj committed 2 years ago

Verified 634b9452

[Traceable Collectives] Hide token for _xla_all_reduce_inplace (#5208)

alanwaketan committed 2 years ago

Verified 1a4e4328

[Functionalization] Remove View in tensor_methods::view (#5212)

alanwaketan committed 2 years ago

Verified 19242d92

Fix set_default_tensor_type usage since it is deprecated (#5211)

JackCaoG committed 2 years ago

Verified 0fcffdc8

Skip test_hybrid_mesh_shape on TPU (#5210)

khatwanimohit committed 2 years ago

Verified 6b93aa95

Enable dynamism for torch.sub (#5193)

vanbasten23 committed 2 years ago

Verified c8503448

Add tesor debugging util API (#5204)

JackCaoG committed 2 years ago

Verified 9e588be1

[Functionalization] Remove View in tensor_methods::slice (#5200)

vanbasten23 committed 2 years ago

Verified c97c40d1

skip test_hybrid_mesh on TPU (#5207)

khatwanimohit committed 2 years ago

Verified a7d42310

[Functionalization] Remove View in tensor_methods::diagonal (#5194)

wonjoo-wj committed 2 years ago

Verified 74960826

[SPMD] Hybrid Device mesh creation (#5147)

khatwanimohit committed 2 years ago

Verified 60a6d606

[OpenXLA] Replace `core:lib` in `util` from TensorFlow to tensorflow/compiler/xla (#5197)

ManfeiBai committed 2 years ago

Verified 0eeaefb2

[OpenXLA] Update `ptxla_cc_test`'s definition using `xla_cc_test` (#5165)

ManfeiBai committed 2 years ago

Verified 06446d83

[OpenXLA] Reimplement Convolution with XLA::Shape (#5190)

ManfeiBai committed 2 years ago

Verified 3d79831c

[Functionalization] Remove views in alias (#5199)

alanwaketan committed 2 years ago

Verified a04b86d7

[Functionalization] Run TestViewOpsXLA without it (#5196)

alanwaketan committed 2 years ago

Verified d0130f36

Implement SPMDSavePlanner to take distributed checkpoints (#5170)

jonb377 committed 2 years ago

Verified 4aa4dcaf

Fix import in SPMD resnet (#5191)

jonb377 committed 2 years ago

Verified 55a2d555

Update nightly build version in triggers (#5192)

will-cromar committed 2 years ago

Verified caf7ea01

[Functionalization] Slightly improve detach_copy (#4814)

alanwaketan committed 2 years ago

Verified 83f186d0

[Functionalization] Remove View in tensor_methods::narrow (#5184)

alanwaketan committed 2 years ago

Verified 39307dc3

[Functionalization] Remove CreateAsStridedViewInfo (#5084)

alanwaketan committed 2 years ago

Verified d4383158

Update package version in ansible build (#5187)

will-cromar committed 2 years ago

Verified ffe1089a

Make sure dynamo give us different graph when input shape changes (#5177)

JackCaoG committed 2 years ago

Verified dfb80361

Support output param sharding in dynamo + spmd (#5101)

wonjoo-wj committed 2 years ago

Verified 84e77567

add tuple remove pass (#5167)

lsy323 committed 2 years ago

Verified 3ca3fd2f

[test][OpenXLA] Delete `core:lib` in torch_xla/csrc/BUILD (#5176)

ManfeiBai committed 2 years ago

Verified 44ff21b9

[test] Delete `core:lib` in `libxla_computation_client.so` (#5175)

ManfeiBai committed 2 years ago

Verified e42365fe

[OpenXLA] Replace `with_tpu_support` from TSL (#5171)

ManfeiBai committed 2 years ago

Verified b4448baa

Move `third_party/xla_client` to `torch_xla/csrc/runtime` (#5160)

will-cromar committed 2 years ago

Verified 0015ea69

Newer Older