pytorch/xla

Pull Requests Commits

Update stablehlo_composite_helper.cc

chunnienc committed 1 year ago

a6f20016

Update test_mark_pattern.py

chunnienc committed 1 year ago

aeec3eda

chunnienc committed 1 year ago

7d5376c5

Merge branch 'master' into chunnienc/hlfb-mutli-outputs

chunnienc committed 1 year ago

8e8c0ab9

Fix model skip yaml file not found issue (#6428)

zpcore committed 1 year ago

Verified e2012881

Revert "Disambiguate `benchmarks` directory search. (#6434)" (#6440)

zpcore committed 1 year ago

Verified bc4fee20

Disambiguate `benchmarks` directory search. (#6434)

ysiraichi committed 1 year ago

Verified 599b9256

update README.md doc to 2.2 (#6437)

zpcore committed 1 year ago

Verified ea9c6481

Avoid fallback for avg_pool - (#6409)

qihqi committed 1 year ago

Verified a0bae821

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

ysiraichi committed 1 year ago

Verified 85c66f5f

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

cota committed 1 year ago

Verified 010b6f09

aggregate.py: add tab format (#6405)

cota committed 1 year ago

Verified 90d9dd1d

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

ysiraichi committed 1 year ago

Verified be44bc6b

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

ysiraichi committed 1 year ago

Verified 492fe27e

Fix some more core aten ops (#6414)

wonjoo-wj committed 1 year ago

Verified a83be44c

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

ysiraichi committed 1 year ago

Verified 0baf1dae

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

alanwaketan committed 1 year ago

Verified ba6a9f5b

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

cota committed 1 year ago

Verified 9334ebbe

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

cota committed 1 year ago

Verified 4fc697cb

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

yeounoh committed 1 year ago

Verified 48a1ae3d

Point to v2.2.0 release (#6418)

zpcore committed 1 year ago

Verified f66406bb

Update OpenXLA pin to 20240124 (#6390)

vanbasten23 committed 1 year ago

Verified 0aee211d

[Pallas] PoC Integration (#6340)

alanwaketan committed 1 year ago

Verified 56db8f23

[FSDPv2] Add a user guide (#6386)

alanwaketan committed 1 year ago

Verified de5d7647

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

cota committed 1 year ago

Verified 2b7f4354

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

ysiraichi committed 1 year ago

Verified 2a17ac20

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

will-cromar committed 1 year ago

Verified bd95eb13

Add compilation time analysis. (#6371)

golechwierowicz committed 1 year ago

Verified 32d24adb

Revert "Lower RandPerm" (#6394)

JackCaoG committed 1 year ago

Verified 2f4275f5

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

cota committed 1 year ago

Verified 259b7dcd

Older