pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

85053c3e

alanwaketan committed 1 year ago

bf274847

Skip tests if not on TPU

alanwaketan committed 1 year ago

29a73c77

alanwaketan committed 1 year ago

6a099a80

alanwaketan committed 1 year ago

7c05c1c4

Extend the input to accepts array

alanwaketan committed 1 year ago

163e0672

Update the payload

alanwaketan committed 1 year ago

27fb56bb

Add missing files

alanwaketan committed 1 year ago

3c1219f5

alanwaketan committed 1 year ago

bae11346

[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

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

cota committed 1 year ago

Verified 77b9b2da

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

cota committed 1 year ago

Verified 9e4db963

Update DDP example in README.md (#6384)

bhavya01 committed 1 year ago

Verified c70e4cc6

Lower RandPerm (#6315)

wonjoo-wj committed 1 year ago

Verified e8b1de97

Add GPU plugin build to CI (#6373)

will-cromar committed 1 year ago

Verified e72cf490

Support PJRT C API `create_options` (#6289)

will-cromar committed 1 year ago

Verified c738156d

Provide op lowering for Logit (#6357)

bhavya01 committed 1 year ago

Verified a4d2d764

Use `benchmark_cls` for checking precision.` (#6375)

ysiraichi committed 1 year ago

Verified a1e51e4c

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

Older