pytorch/xla

Pull Requests Commits

Codegen acos and acosh

wonjoo-wj committed 3 years ago

a0c4ec13

remove comments

JackCaoG committed 3 years ago

49ec0e9a

Remove node_ctor_args_call

JackCaoG committed 3 years ago

58277103

Codegen Abs and Maximum op

JackCaoG committed 3 years ago

039102da

Shift int only (#3561)

JackCaoG committed 3 years ago

Verified 344c1b1d

Update for consolidated contiguous/sizes policy (#3555)

ezyang committed 3 years ago

Verified bc310a94

Implement Fully Sharded Data Parallel (FSDP) in PyTorch XLA (#3431)

ronghanghu committed 3 years ago

Verified 3c83269f

allow disabling layout pinning in optimizer_step and reduce_gradients (#3556)

ronghanghu committed 3 years ago

Verified 6e3992ac

Lower LogSigmoid, LogSigmoidBackward, Elu, and EluBackward (#3539)

wonjoo-wj committed 3 years ago

Verified cc312e27

Delete .torch_pin

JackCaoG committed 3 years ago

Verified 8baddaba

Remove unpool backward (#3537)

JackCaoG committed 3 years ago

Verified 87fda0ac

Update matmul cpp test to check a different counter (#3154)

JackCaoG committed 3 years ago

Verified aa34578e

Lower celu, celu_, selu, and selu_ (#3547)

wonjoo-wj committed 3 years ago

Verified d1de2477

Add ir_builder include since MakeNode moved to this file (#3548)

JackCaoG committed 3 years ago

Verified a3031d64

Start implementation of PjRtComputationClient. (#3463)

will-cromar committed 3 years ago

Verified 9525a8ec

Fix view with no-op view_info (#3540)

JackCaoG committed 3 years ago

Verified c8661df4

change new line to "\n" in XLA_EMIT_STEPLOG (#3546)

ronghanghu committed 3 years ago

Verified 6cc9f3f7

Remove namespaces ir and ops (#3533)

wonjoo-wj committed 3 years ago

Verified 9df591d7

Update libtpu nightly version to 0.1.dev20220413 (#3541)

yeounoh committed 3 years ago

Verified 4131f1ec

Tf update 04 2022 (#3535)

yeounoh committed 3 years ago

Verified bb1e5489

Rotate the nvidia repo key in the base docker image (#3538)

yeounoh committed 3 years ago

Verified f684fd43

Move GetXlaTensorOrCreateForWrappedNumber (#3534)

JackCaoG committed 3 years ago

Verified facc7126

Revert "Start XRT server in a separate process in CircleCI CPU test (#3519)" (#3536)

JackCaoG committed 3 years ago

Verified 093d443b

Move device lock before the execution instead of tensor gathering (#3457)

JackCaoG committed 3 years ago

Verified 5d1c4210

Start XRT server in a separate process in CircleCI CPU test (#3519)

JackCaoG committed 3 years ago

Verified 935b6024

Remove TPU_HOST_BOUNDS setting for TPU single host (#3525)

JackCaoG committed 3 years ago

Verified 9f9db76a

Lower Gelu and GeluBackward (#3531)

wonjoo-wj committed 3 years ago

Verified 89c5dad4

Rename torch_xla::ir Node to XlaNode and Value to XlaValue (#3526)

wonjoo-wj committed 3 years ago

Verified d9dc45fc

Add lazy::shape and lazy::shapefn to XlaNode constructor with some clean up (#3528)

JackCaoG committed 3 years ago

Verified cf7e123c

lowering roll op (#3505)

cicirori committed 3 years ago

Verified 4f6b2833

Older