pytorch/xla

Pull Requests Commits

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

Codegen take (#3818)

wonjoo-wj committed 3 years ago

Verified 1715ccd0

Codegen ne (#3882)

JackCaoG committed 3 years ago

Verified 0064f439

codegen eq (#3879)

JackCaoG committed 3 years ago

Verified c91bf61d

Use tfrt cpu client (#3898)

darisoy committed 3 years ago

Verified a6ea0f34

Update `isDynamic` api to align with `is_symbolic` API in upstream (#3888)

miladm committed 3 years ago

Verified 35e6c49b

upgrading libtpu version to jun. 23 to align with that of tensorflow (#3899)

miladm committed 3 years ago

Verified dd88cc52

Codegen all.dim (#3865)

wonjoo-wj committed 3 years ago

Verified accd86af

Codegen elu and elu_ (#3893)

wonjoo-wj committed 3 years ago

Verified 76df1300

use XLA patched linear in FSDP (fix #3811 and #3718) and expose options on param sharding dim and pinning memory (#3830)

ronghanghu committed 3 years ago

Verified 3f47abab

Experimental TPU Pod support with PjRt (#3813)

will-cromar committed 3 years ago

Verified 63c67dfc

Codegen clamp.Tensor (#3869)

wonjoo-wj committed 3 years ago

Verified 9f7dfe22

Enable `SetUpperBound` to update DS upper bounds (#3890)

miladm committed 3 years ago

Verified cf337f09

Codegen lt le (#3876)

JackCaoG committed 3 years ago

Verified 36583d46

Codegen addcdiv and addcmul (#3768)

JackCaoG committed 3 years ago

Verified 6bfcd242

Newer Older