pytorch/xla

Pull Requests Commits

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

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

Newer Older