pytorch/xla

Pull Requests Commits

Update test_core_aten_ops.py

chunnienc committed 2 years ago

e9b4c8e9

Enable and add aten logical and tests

chunnienc committed 2 years ago

6613bc15

Update example Kaggle notebook to PyTorch/XLA 2.1 (#6287)

will-cromar committed 2 years ago

Verified 0e735dec

Add CPU PJRT plugin for testing (#6253)

will-cromar committed 2 years ago

Verified 2f1334d7

Temporarily disable GPU coverage workflow in CI (#6285)

yeounoh committed 2 years ago

Verified 83b7571a

Backport `PjRtStreamExecutorLoadedExecutable:GetCompileOptions` (#6282)

will-cromar committed 2 years ago

Verified 235b82b5

Remove or simplify hardcoded lists of device types (#6235)

will-cromar committed 2 years ago

Verified 050a2403

Adapt a few `torch.utils.checkpoint` functions for PyTorch/XLA. (#6178)

ysiraichi committed 2 years ago

Verified ebb200b0

Limit number of bazel jobs in the build (#6279)

will-cromar committed 2 years ago

Verified 9a4ef68a

Revert "Remove reshape lowering (#6230)" (#6280)

cota committed 2 years ago

Verified c3a59095

Fix a typo in aten_xla_type.cpp (#6271)

bhavya01 committed 2 years ago

Verified f2416cf8

Revert "remove model training using openxla backend (#6262)" (#6276)

zpcore committed 2 years ago

Verified 551c0dfa

Add matmul microbenchmark. (#6259)

golechwierowicz committed 2 years ago

Verified 88d97cc7

Measure GPU wall time without launch overheads. (#6233)

golechwierowicz committed 2 years ago

Verified bd82beb9

Set min TPU nodes to 0 for TPU CI & disable workflow run on push (#6270)

mbzomowski committed 2 years ago

Verified 257f0f53

test_zero1 passes on GPU with latest torch/xla (commit e1c94df) (#6268)

jeffhataws committed 2 years ago

Verified a60f8e7c

remove model training using openxla backend (#6262)

zpcore committed 2 years ago

Verified a89ebefd

Add workflow file for TPU CI (#6198)

mbzomowski committed 2 years ago

Verified 5801f7e5

Remove skip from prod lowering test (#6265)

LukeBoyer committed 2 years ago

Verified e1c94dfa

Remove skip from logical_or lowering test (#6264)

LukeBoyer committed 2 years ago

Verified 9eaab55a

Set all_reduce_token to None when exiting (#6247)

yitongh committed 2 years ago

Verified f9c12fc1

Set devices argument in dist.Backend.register_backend call. (#6254)

vanbasten23 committed 2 years ago

Verified 99cfeee6

Experimental GPU PJRT Plugin (#6240)

will-cromar committed 2 years ago

Verified 29977cbd

Remove obsolete device special cases (#6197)

will-cromar committed 2 years ago

Verified 0cd6f105

update release rc wheel name to RC6 (#6245)

ManfeiBai committed 2 years ago

Verified d96c15d6

Remove reshape lowering (#6230)

JackCaoG committed 2 years ago

Verified 0c81365e

Fix some core aten ops (#6243)

wonjoo-wj committed 2 years ago

Verified d7c44308

Create sharded data placeholders directly (#6203)

will-cromar committed 2 years ago

Verified efa6fcfd

Remove Colab notebooks (#6155)

will-cromar committed 2 years ago

Verified 26dcba94

benchmarks: add nightly.sh to automate regular benchmarking (#6231)

cota committed 2 years ago

Verified 99fdf9a2

Older