pytorch/xla

Pull Requests Commits

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

Remove `test_experimental_pjrt_tpu.py` from TPU CI (#5247)

will-cromar committed 2 years ago

Verified c9f2d91a

remove duplicate autocast_test (#5246)

cowanmeg committed 2 years ago

Verified 458cdce2

AMP for TPUs v3 (#5161)

cowanmeg committed 2 years ago

Verified 347146ec

Only run the main CI workflow on PRs targeting master and release branches (#5244)

will-cromar committed 2 years ago

Verified 9233464d

Add function to generate stablehlo based callable from pytorch model (#5216)

qihqi committed 2 years ago

Verified 13526603

Add test workflow for `xrt` branch (#5241)

will-cromar committed 2 years ago

Verified 025a2750

Remove legacy file reading APIs (#5234)

will-cromar committed 2 years ago

Verified 1dc61564

[Functionalization] Remove View in tensor_methods::transpose (#5198)

wonjoo-wj committed 2 years ago

Verified b3f4c7bc

Enable source line info in StableHLO (#5203)

lsy323 committed 2 years ago

Verified 7d8baf63

Don't let autograd initialize the ComputationClient. (#5068)

will-cromar committed 2 years ago

Verified 97e12ea3

[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

Newer Older