pytorch/xla

Pull Requests Commits

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

Implement SPMDLoadPlanner to enable distributed checkpoint loading (#5130)

jonb377 committed 2 years ago

Verified 50821dac

Refine the comment for TensorCollectionBarrier in SyncTensorsGraphInternal (#5166)

JackCaoG committed 2 years ago

Verified 36536a7a

[OpenXLA] Replace `tf_cc_shared_object` with `cc_binary` (#5153)

ManfeiBai committed 2 years ago

Verified 4f66d6d8

Lower count_nonzero (#5137)

vanbasten23 committed 2 years ago

Verified 4da677c7

Throw an error when both operands of torch.add have dynamic dimensions (#5157)

vanbasten23 committed 2 years ago

Verified 464b967a

Add utility functions for distributed checkpointing (#5128)

jonb377 committed 2 years ago

Verified c7fe0f98

Fix alias for deprecated `pjrt.rendezvous` (#5155)

will-cromar committed 2 years ago

Verified 50431339

Revert "AMP for TPUs v2 (#5148)" (#5154)

JackCaoG committed 2 years ago

Verified a4500e74

Newer Older