pytorch/xla

Pull Requests Commits

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

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

JackCaoG committed 3 years ago

Verified 8f1f13af

[PjRt] Implement `xm.rendezvous` for TPU v4. (#3929)

will-cromar committed 3 years ago

Verified 5a09efdf

Add lowring for floor_divide (#3923)

ymwangg committed 3 years ago

Verified 28b45da5

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

ezyang committed 3 years ago

Verified 629ac43a

Codegen any and any.dim (#3910)

wonjoo-wj committed 3 years ago

Verified 5df0fbf4

Codegen frac op (#3930)

wonjoo-wj committed 3 years ago

Verified 227ac13d

Revert "Revert "Companion PR for #80074 (#3918)" (#3931)" (#3933)

JackCaoG committed 3 years ago

Verified d8db50a7

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

JackCaoG committed 3 years ago

Verified cd09e19f

Revert "Companion PR for #80074 (#3918)" (#3931)

JackCaoG committed 3 years ago

Verified cae9db29

Set PJRT device assignment in order of device ID (#3902)

will-cromar committed 3 years ago

Verified 1e0bb990

Companion PR for #80074 (#3918)

lezcano committed 3 years ago

Verified f82e5d8a

Companion PR for https://github.com/pytorch/pytorch/pull/83628 (#3914)

ezyang committed 3 years ago

Verified 3508e9db

Codegen for bitwise and, or, xor, and not (#3815)

steventk-g committed 3 years ago

Verified 1f8a7ee1

Fix upstream break 83586 (#3912)

JackCaoG committed 3 years ago

Verified 78e80924

Codegen cholesky (#3907)

wonjoo-wj committed 3 years ago

Verified 4d85d36f

Fix shallowCopy logical_element type issue (#3911)

JackCaoG committed 3 years ago

Verified 9b9b39cb

Newer Older