pytorch/xla

Pull Requests Commits

remove size base

JackCaoG committed 2 years ago

307fc2e0

Update spmd doc

JackCaoG committed 2 years ago

2e85f400

Add FSDPv2 example for the decoder only model (#7088)

JackCaoG committed 2 years ago

Verified f336317e

Add torch_xla2 `export_program_to_stablehlo` API with unbounded dynamism support (#7093)

GleasonK committed 2 years ago

Verified baf08aea

Add dlpack support (#7025)

vanbasten23 committed 2 years ago

Verified 60238557

add missing aten op (#7078)

zpcore committed 2 years ago

Verified 5e1d454e

Add data-type promotion to `gelu_backward`. (#7090)

ysiraichi committed 2 years ago

Verified 8d35eb05

Dump HLO HBM usage info (#7085)

JackCaoG committed 2 years ago

Verified 206f1b7f

Implement `ComputationClient::GetMemoryInfo` (#7086)

will-cromar committed 2 years ago

Verified 5409cd5b

add example for fsdp (#7061)

JackCaoG committed 2 years ago

Verified 961c22ae

Add example for a decoder only model (#7082)

JackCaoG committed 2 years ago

Verified e0fb8782

Fix usage of extract_jax (#7075)

zpcore committed 2 years ago

Verified e0d5a49a

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

ysiraichi committed 2 years ago

Verified a2540acb

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

ysiraichi committed 2 years ago

Verified 8990f1b9

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

will-cromar committed 2 years ago

Verified 3c59087e

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

will-cromar committed 2 years ago

Verified 56c2368c

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

qihqi committed 2 years ago

Verified 8247aec1

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

lsy323 committed 2 years ago

Verified a6ee8a50

format the output model input (#6869)

zpcore committed 2 years ago

Verified 68daf61f

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

ysiraichi committed 2 years ago

Verified aeed89eb

Support torch_xla2 benchmarking using torchbench (#7013)

zpcore committed 2 years ago

Verified 9e189350

add DDP with SPMD example (#7063)

JackCaoG committed 2 years ago

Verified c6074abd

Update TPU CI debugging tips (#7066)

mbzomowski committed 2 years ago

Verified a8eae0d9

add amp example (#7062)

JackCaoG committed 2 years ago

Verified 1fa1f858

reenable disabled pt2e test (#7059)

lsy323 committed 2 years ago

Verified df0d147e

Support megacore_mode in paged_attention (#7060)

wonjoo-wj committed 2 years ago

Verified cbb9e213

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

mbzomowski committed 2 years ago

Verified f26c35c2

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

JackCaoG committed 2 years ago

Verified ae63cd1e

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

qihqi committed 2 years ago

Verified b64d8a2a

Pin update 20240513 (#7052)

qihqi committed 2 years ago

Verified 2b28ae29

Older