pytorch/xla

Pull Requests Commits

Don't load libtpu on deployment_tests

JackCaoG committed 3 years ago

94152e5a

Make SizeNode to use XlaShape and add test (#3954)

JackCaoG committed 3 years ago

Verified e9b8d435

Making sure XLA_HLO_DEBUG populate the scope metadata (#3985)

JackCaoG committed 3 years ago

Verified 0aed7b64

Fix special scalar handling for addcdiv and addcmul (#3953)

ymwangg committed 3 years ago

Verified 88d934e3

Set three_fry as the default RNG for GPU (#3951)

ymwangg committed 3 years ago

Verified 2d877167

Create README.md for CircleCI (#3916)

wonjoo-wj committed 3 years ago

Verified 2bca9530

Disable audio build (#3989)

JackCaoG committed 3 years ago

Verified 7a0d4958

Companion PR for https://github.com/pytorch/pytorch/pull/84837 (#3987)

ezyang committed 3 years ago

Verified 8c2f9e75

Dsiable cpp test build and run to unblock nightly wheel build (#3986)

JackCaoG committed 3 years ago

Verified e0dcc317

Make `ExpectCounterChanged` to account for `_symint` ops without fixing the tests or ops explicitly (#3978)

Krovatkin committed 3 years ago

Verified 1e71c8e9

Add ViT model to the supported models in the tests. (#3977)

vanbasten23 committed 3 years ago

Verified 6a11face

Remove unnecessary virtual method. (#3981)

ezyang committed 3 years ago

Verified 5f7d4164

[PJRT] Remove thread-local replica ordinals (#3973)

will-cromar committed 3 years ago

Verified 28ea3758

[DDP] Add a correctness test case (#3980)

alanwaketan committed 3 years ago

Verified 16fa65f4

[SPMD][PoC] Tensor sharding util (#3962)

yeounoh committed 3 years ago

Verified afb00b06

enable test/nn/test_pooling.py (#3971)

kshitij12345 committed 3 years ago

Verified b7f54864

Fix DDP initialization with XLA backend (#3952)

will-cromar committed 3 years ago

Verified 6e42e7cb

Make expand to support is_alias_of (#3967)

alanwaketan committed 3 years ago

Verified bbc7878b

Revert "Add libtpu configuration to cpp test script (#3957)" (#3970)

JackCaoG committed 3 years ago

Verified b8688ee3

run the new tests under `test/nn` (#3955)

kshitij12345 committed 3 years ago

Verified 11511b35

Support Tensor.is_alias_of (#3966)

alanwaketan committed 3 years ago

Verified 1f53bae4

Fix the location of libxla_computation_client.so (#3964)

alanwaketan committed 3 years ago

Verified 50032d21

Replace TPU_VISIBLE_DEVICES with TPU_VISIBLE_CHIPS (#3956)

jonb377 committed 3 years ago

Verified 06fc573d

Clean up SymIntElements (#3944)

JackCaoG committed 3 years ago

Verified f77da332

Add libtpu configuration to cpp test script (#3957)

steventk-g committed 3 years ago

Verified bd1a5a41

Revert "revert numel companion PR to 84207" (#3950)

Krovatkin committed 3 years ago

Verified 5059c3b1

Add regression test for torch.expand (#3785)

ymwangg committed 3 years ago

Verified 205ae574

revert numel companion PR to 84207 (#3948)

Krovatkin committed 3 years ago

Verified d1d693bc

Revert "Revert "Revert "Revert "Companion PR for https://github.com/pytorch/pytorch/pull/83628 (#3914)" (#3932)" (#3938)" (#3945)" (#3946)

ezyang committed 3 years ago

Verified 2f63f7e9

Helper function to wrap entry HLO (#3920)

JackCaoG committed 3 years ago

Verified 41a4f855

Older