pytorch/xla

Pull Requests Commits

Dsiable dropout2d test

JackCaoG committed 3 years ago

e4ac0f6d

JackCaoG committed 3 years ago

b8aa84c9

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

TensorFlow June pin update & PyTorch/XLA c++17 migration (follow up) (#3794)

miladm committed 3 years ago

Verified 6639bcc4

Add profiler traces to `PjRtComputationClient` (#3866)

will-cromar committed 3 years ago

Verified 48bf0fde

Add env var to disable the neumeric token that is used to string CC o… (#3825)

hjm-aws committed 3 years ago

Verified 3935e444

Codegen amin amax (#3771)

JackCaoG committed 3 years ago

Verified b8cef3e4

Codegen clamp_min.Tensor and clamp_max.Tensor (#3867)

wonjoo-wj committed 3 years ago

Verified 3eb8a9d9

Codegen ge, gt (#3864)

JackCaoG committed 3 years ago

Verified 728b3523

Remove Ger lowering as it is not needed (#3855)

JackCaoG committed 3 years ago

Verified 2d880d15

Codegen op and rename existing op to all_dim (#3856)

wonjoo-wj committed 3 years ago

Verified 45689b9f

Update codegen doc with more example (#3847)

JackCaoG committed 3 years ago

Verified 58d2ba02

added TPU_ML_PLATFORM variable (#3846)

miladm committed 3 years ago

Verified 987d5405

Refactor PjRt device transfers to use `BufferFromHostBuffer` (#3833)

will-cromar committed 3 years ago

Verified 5dfd843c

partially codegen adaptive_avgpool3d and backward (#3790)

JackCaoG committed 3 years ago

Verified 1f154ceb

Add parameter broadcasting to PJRT examples. (#3836)

darisoy committed 3 years ago

Verified 477ca244

Respect default dtype (#3798)

peterbell10 committed 3 years ago

Verified 7b6747df

Revert "Revert "Revert of #3822 "Revert "implementing sym_numel_custom"" (#3826)" (#3835)" (#3837)

Krovatkin committed 3 years ago

Verified 774df50c

Revert "Revert of #3822 "Revert "implementing sym_numel_custom"" (#3826)" (#3835)

Krovatkin committed 3 years ago

Verified 3af08737

remove pin (#3834)

Krovatkin committed 3 years ago

Verified 2c446ff1

Revert of #3822 "Revert "implementing sym_numel_custom"" (#3826)

Krovatkin committed 3 years ago

Verified ce6b404c

Support dim=None for std (#3792)

wonjoo-wj committed 3 years ago

Verified 5d294a9e

Revert "implementing sym_numel_custom" (#3822)

Krovatkin committed 3 years ago

Verified ff2e0ea5

implementing sym_numel_custom (#3784)

Krovatkin committed 3 years ago

Verified b358d985

Older