pytorch/xla

Pull Requests Commits

ManfeiBai committed 2 years ago

Verified beeda05e

Update pip.yaml

ManfeiBai committed 2 years ago

Verified 032cb246

Update Dockerfile

ManfeiBai committed 2 years ago

Verified 0782c585

update README.md doc to 2.2 (#6437)

zpcore committed 2 years ago

Verified ea9c6481

Avoid fallback for avg_pool - (#6409)

qihqi committed 2 years ago

Verified a0bae821

Fix distributed checkpoint errors when PyTorch CUDA is enabled. (#6421)

ysiraichi committed 2 years ago

Verified 85c66f5f

Revert "aggregate.py: add tab format" (#6424)

cota committed 2 years ago

Verified 010b6f09

aggregate.py: add tab format (#6405)

cota committed 2 years ago

Verified 90d9dd1d

[benchmarks] Fix CUDA device check. (#6419)

ysiraichi committed 2 years ago

Verified be44bc6b

Use PyTorch's dynamo benchmark skip-list. (#6416)

ysiraichi committed 2 years ago

Verified 492fe27e

Fix some more core aten ops (#6414)

wonjoo-wj committed 2 years ago

Verified a83be44c

[torchbench] Re-land: Force `llama_v2_7b_16h` to use SGD. (#6402)

ysiraichi committed 2 years ago

Verified 0baf1dae

Move torch import closer to the place it uses (#6412)

alanwaketan committed 2 years ago

Verified ba6a9f5b

nightly.sh: pass --always to git describe --tags (#6406)

cota committed 2 years ago

Verified 9334ebbe

nightly.sh: add flag to set Hugging Face token (#6410)

cota committed 2 years ago

Verified 4fc697cb

Revert: #6368 to fix nightly CI coverage run. (#6369)

yeounoh committed 2 years ago

Verified 48a1ae3d

Point to v2.2.0 release (#6418)

zpcore committed 2 years ago

Verified f66406bb

Update OpenXLA pin to 20240124 (#6390)

vanbasten23 committed 2 years ago

Verified 0aee211d

[Pallas] PoC Integration (#6340)

alanwaketan committed 2 years ago

Verified 56db8f23

[FSDPv2] Add a user guide (#6386)

alanwaketan committed 2 years ago

Verified de5d7647

aggregate.py: allow picking backends to report on (#6392)

cota committed 2 years ago

Verified 2b7f4354

Force to fp16 models to fp32 if `XLA_USE_FP16` is already set. (#6389)

ysiraichi committed 2 years ago

Verified 2a17ac20

Define PJRT plugin interface in C++ (#6360)

will-cromar committed 2 years ago

Verified bd95eb13

Add compilation time analysis. (#6371)

golechwierowicz committed 2 years ago

Verified 32d24adb

Revert "Lower RandPerm" (#6394)

JackCaoG committed 2 years ago

Verified 2f4275f5

benchmarks/nightly.sh: add -P flag to enable profiling (#6298)

cota committed 2 years ago

Verified 259b7dcd

aggregate.py: add --baseline option (#6349)

cota committed 2 years ago

Verified 77b9b2da

Reapply "Fix some more core aten ops (#6342)" (#6377) (#6387)

cota committed 2 years ago

Verified 9e4db963

Update DDP example in README.md (#6384)

bhavya01 committed 2 years ago

Verified c70e4cc6

Lower RandPerm (#6315)

wonjoo-wj committed 2 years ago

Verified e8b1de97

Older