pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

db141fe2

alanwaketan committed 1 year ago

9ea14ef8

alanwaketan committed 1 year ago

5e7ce0f1

Move maxpool to the original place

alanwaketan committed 1 year ago

cef020f5

Replace quantize

alanwaketan committed 1 year ago

bb075e36

alanwaketan committed 1 year ago

e3b9ce14

[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

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

Older