pytorch/xla

Pull Requests Commits

Revert "Add `AllReduceScatter` XLA collective op (#3075)" (#3088)

JackCaoG committed 4 years ago

Verified e9345161

Add `AllReduceScatter` XLA collective op (#3075)

jysohn23 committed 4 years ago

Verified bc4aee9b

Add extra_require for TPU VM libtpu installation (#3073)

jysohn23 committed 4 years ago

Verified 81eecf45

Bump up time limit for test_async_closures (#3066)

JackCaoG committed 4 years ago

Verified 85b8ad1b

Start "Common Issues" section in troubleshooting (#3069)

jysohn23 committed 4 years ago

Verified 74067733

Enabling trace requests to retry loop until timeout (#3067)

jysohn23 committed 4 years ago

Verified 1de1a3f7

Update logical_and opkind (#3064)

JackCaoG committed 4 years ago

Verified 7f1f1039

Add amp flag to imagenet test for GPU (#3059)

JackCaoG committed 4 years ago

Verified e0c51aec

Disable test that fails on TPU (#3060)

JackCaoG committed 4 years ago

Verified 59739f2e

Add AMP test (#3057)

JackCaoG committed 4 years ago

Verified 393cb707

Disable test_Conv2d_backward_depthwise_xla_float64 (#3058)

JackCaoG committed 4 years ago

Verified 4e558b48

Fix profiler notebook stuck on "Loading Data" (#3056)

jysohn23 committed 4 years ago

Verified 88474a32

Add `SyncBatchNorm` (#3048)

jysohn23 committed 4 years ago

Verified c7647651

Lowering for LogicalAnd (#3054)

JackCaoG committed 4 years ago

Verified 9146a542

[reland] #3019: [test] add opinfo based eager reference test (#3053)

kshitij12345 committed 4 years ago

Verified 5fba0dbc

Revert "[test] add opinfo based eager reference test (#3019)" (#3051)

JackCaoG committed 4 years ago

Verified c297289b

[test] add opinfo based eager reference test (#3019)

kshitij12345 committed 4 years ago

Verified f18f94bb

initial implementation of amax. BuildAMax is TBD. Test is TBD (#3027)

miladm committed 4 years ago

Verified 30e94a24

Handle bitwise_ops(XLATensor, CpuTensor) and allow Pred->Uint8 promotion (#3022)

JackCaoG committed 4 years ago

Verified 2ad14e1f

parent db950f7fb7fcdcb69958ebb9c2a8b9ece05ba052 (#3034)

miladm committed 4 years ago

Verified 63b4bcb7

Do not always enable verbose for bazel build (#3040)

JackCaoG committed 4 years ago

Verified db950f7f

hyperlink fix (#3038)

miladm committed 4 years ago

Verified e40dab21

improved the op lowering guideline (#3037)

miladm committed 4 years ago

Verified a0dd9067

Add some loging to make sync and get tensor more clear (#3036)

JackCaoG committed 4 years ago

Verified 132255a3

Fix VarMean node kind

asuhan committed 4 years ago

5ce99bff

Disable test_cov on TPU due to precision issue (#3032)

JackCaoG committed 4 years ago

Verified 31a26de4

Make kernels methods (#3012)

bdhirsh committed 4 years ago

Verified d033f8d3

fix transpose_ (#3018)

bdhirsh committed 4 years ago

Verified 4eedb650

Add fp16 support for ARange op (#3026)

luchangli03 committed 4 years ago

Verified 2f04ea39

Add expecttest (#3023)

JackCaoG committed 4 years ago

Verified 86998fe3

Newer Older