pytorch/xla

Pull Requests Commits

JackCaoG committed 1 year ago

694fde66

JackCaoG committed 1 year ago

a05f5a10

Append graph memory usage info for PT_XLA_DEBUG=1

JackCaoG committed 1 year ago

e4b1a7e1

Dump HLO HBM usage info

JackCaoG committed 1 year ago

b4d83e43

Add example for a decoder only model (#7082)

JackCaoG committed 1 year ago

Verified e0fb8782

Fix usage of extract_jax (#7075)

zpcore committed 1 year ago

Verified e0d5a49a

Fix overflow for `div` arguments. (#7081)

ysiraichi committed 1 year ago

Verified a2540acb

[benchmarks] Add default value to `move_to_device`. (#7080)

ysiraichi committed 1 year ago

Verified 8990f1b9

Add `xla.step` context manager (#7068)

will-cromar committed 1 year ago

Verified 3c59087e

Add VSCode devcontainer instructions to CONTRIBUTING.md (#7072)

will-cromar committed 1 year ago

Verified 56c2368c

Add a unit test for MoE layer. (#7069)

qihqi committed 1 year ago

Verified 8247aec1

Map jnp.int4 to torch.int8 (#7071)

lsy323 committed 1 year ago

Verified a6ee8a50

format the output model input (#6869)

zpcore committed 1 year ago

Verified 68daf61f

[benchmarks] Fix AMP setup for torchbench models. (#7067)

ysiraichi committed 1 year ago

Verified aeed89eb

Support torch_xla2 benchmarking using torchbench (#7013)

zpcore committed 1 year ago

Verified 9e189350

add DDP with SPMD example (#7063)

JackCaoG committed 1 year ago

Verified c6074abd

Update TPU CI debugging tips (#7066)

mbzomowski committed 1 year ago

Verified a8eae0d9

add amp example (#7062)

JackCaoG committed 1 year ago

Verified 1fa1f858

reenable disabled pt2e test (#7059)

lsy323 committed 1 year ago

Verified df0d147e

Support megacore_mode in paged_attention (#7060)

wonjoo-wj committed 1 year ago

Verified cbb9e213

Update runner and runner-container-hooks versions (#7058)

mbzomowski committed 1 year ago

Verified f26c35c2

Add simple example for how to use torch_xla (#7048)

JackCaoG committed 1 year ago

Verified ae63cd1e

Add example to support pytorch lightning; misc bug fixes (#7054)

qihqi committed 1 year ago

Verified b64d8a2a

Pin update 20240513 (#7052)

qihqi committed 1 year ago

Verified 2b28ae29

Add option to export FX Node metadata to StableHLO (#7046)

lsy323 committed 1 year ago

Verified 6f392ccc

[Pallas] Add test_pallas_spmd.py to tpu ci (#7045)

alanwaketan committed 1 year ago

Verified 5cb473a2

[FSDPv2] Fix test_fsdp_v2_multi_slice (#7055)

alanwaketan committed 1 year ago

Verified 4a1588c3

[FSDPv2] Support MultiSlice (#7044)

alanwaketan committed 1 year ago

Verified 6f0b61e5

Adding megablox gmm standalone (#6940)

miladm committed 1 year ago

Verified 40f7e1f5

add a metrics to track persistent cache loading time (#7039)

JackCaoG committed 1 year ago

Verified c1b745e5

Older