pytorch/xla

Pull Requests Commits

Create fori_loop.py

ManfeiBai committed 1 year ago

Verified d6014d43

Create fori_loop.py

ManfeiBai committed 1 year ago

Verified 9ca0be73

Fix walrus operator precedence. (#6374)

ysiraichi committed 1 year ago

Verified 44660d8b

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

wonjoo-wj committed 1 year ago

Verified 4ab7a249

Add documentation around CheckpointManager (#6356)

jonb377 committed 1 year ago

Verified 15213160

disable some models for torchbench (#6352)

zpcore committed 1 year ago

Verified a9005e3c

[Core Aten Ops] Add test_aten_index_put back (#6370)

ManfeiBai committed 1 year ago

Verified 2a5ad315

Update spmd.md with SPMD debug tool (#6358)

ManfeiBai committed 1 year ago

Verified bc2ebed8

fix core aten asinh (#6365)

zpcore committed 1 year ago

Verified 07832b07

Revert "Fix coverage CI run bug & re-enable the EmbeddingBag test with 1% tol" (#6368)

yeounoh committed 1 year ago

Verified c4b5ab6e

remove random number op aten_randn in the test (#6364)

zpcore committed 1 year ago

Verified 566f3d08

Fix coverage CI run bug & re-enable the EmbeddingBag test with 1% tol (#6361)

yeounoh committed 1 year ago

Verified aa42fee0

Fix some more core aten ops (#6342)

wonjoo-wj committed 1 year ago

Verified 99a1341b

[torchbench] Force `llama_v2_7b_16h` to use SGD. (#6350)

ysiraichi committed 1 year ago

Verified 077b10c1

Improve build process for PJRT plugins (#6314)

will-cromar committed 1 year ago

Verified 61bd1d23

[SPMD] update test format (#6355)

ManfeiBai committed 1 year ago

Verified 269ebcf8

fix issue that torchbench uses xla as device instead of tpu (#6341)

zpcore committed 1 year ago

Verified 0576fe95

Export no-longer accepts callable; wrap in nn.Module. (#6345)

qihqi committed 1 year ago

Verified 1c22fe3a

disable test_EmbeddingBag_per_sample_weights_and_new_offsets (#6351)

JackCaoG committed 1 year ago

Verified e265fd7b

Update README.md for cicleci (#6354)

wonjoo-wj committed 1 year ago

Verified 2e7a84b8

Use SGD for list of models from PyTorch. (#6324)

ysiraichi committed 1 year ago

Verified a7918a72

Allow xla bloat16 torch test results with some variances (#6338)

yeounoh committed 1 year ago

Verified ab28b23f

Update test_spmd_debugging.py to avoid code test code self (#6263)

ManfeiBai committed 1 year ago

Verified df097d75

Update pytorch_test_base.py (#6337)

wonjoo-wj committed 1 year ago

Verified ec23c917

Clean-up CUDA cache. (#6325)

ysiraichi committed 1 year ago

Verified 423bb0b2

Raise an error in `xmp.spawn` if runtime is initialized (#6291)

will-cromar committed 1 year ago

Verified e93597de

Update metric check for resnet18 dynamo unit test (#6307)

wonjoo-wj committed 1 year ago

Verified 8c5e2897

Add option to register installed plugins automatically (#6328)

will-cromar committed 1 year ago

Verified d7ddecdb

Revert "Set all_reduce_token to None when exiting" (#6321)

vanbasten23 committed 1 year ago

Verified f33634e4

Increase absolute tolerance for f16 aten_addmm test (#6332)

LukeBoyer committed 1 year ago

Verified 151dc1d8

Older