pytorch/xla

Pull Requests Commits

ManfeiBai committed 1 year ago

Verified 8367f997

enable GPU test

ManfeiBai committed 1 year ago

Verified fc91e8b8

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 7eb80d83

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 01488e3b

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified b05a7396

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified ffd19f6e

ManfeiBai committed 1 year ago

Verified 057397e2

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 2215e9d9

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified e0434985

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified b2444c9d

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 09939110

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 94ae8235

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 90cdfd22

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 89973700

ManfeiBai committed 1 year ago

Verified 0271290e

Merge branch 'master' into ManfeiBai-patch-54

ManfeiBai committed 1 year ago

Verified 1fedc5f3

Temporarily disable GPU coverage workflow in CI (#6285)

yeounoh committed 1 year ago

Verified 83b7571a

Backport `PjRtStreamExecutorLoadedExecutable:GetCompileOptions` (#6282)

will-cromar committed 1 year ago

Verified 235b82b5

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

will-cromar committed 1 year ago

Verified 050a2403

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

ysiraichi committed 1 year ago

Verified ebb200b0

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

will-cromar committed 1 year ago

Verified 9a4ef68a

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

cota committed 1 year ago

Verified c3a59095

Fix a typo in aten_xla_type.cpp (#6271)

bhavya01 committed 1 year ago

Verified f2416cf8

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

zpcore committed 1 year ago

Verified 551c0dfa

Add matmul microbenchmark. (#6259)

golechwierowicz committed 1 year ago

Verified 88d97cc7

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

golechwierowicz committed 1 year ago

Verified bd82beb9

Update test_spmd_debugging.py

ManfeiBai committed 1 year ago

Verified 68cb4208

test for GPU too

ManfeiBai committed 1 year ago

Verified f0351f6d

Update test_spmd_debugging.py with clean sharding string

ManfeiBai committed 1 year ago

Verified 7a7989c0

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

mbzomowski committed 1 year ago

Verified 257f0f53

Older