pytorch/xla

Pull Requests Commits

Output 1st element

alanwaketan committed 1 year ago

c65be9fb

[Pallas] Make gmm functional (#7117)

alanwaketan committed 1 year ago

Verified a9b4fadf

Update XLA pin to 2024/05/24 and fix Hermetic Python integration (#7110)

vam-google committed 1 year ago

Verified 7d31f7de

[Pallas] Support _histogram (#7115)

alanwaketan committed 1 year ago

Verified cb805837

implement Repeat with fixed output shape (#7114)

JackCaoG committed 1 year ago

Verified 3369bf7d

[Pallas] Introduce _make_group_metadata (#7107)

alanwaketan committed 1 year ago

Verified 22e912ea

fix jax dependency bug (#7105)

zpcore committed 1 year ago

Verified cb8533be

Add data-type promotion to `stack`. (#7091)

ysiraichi committed 1 year ago

Verified a299f337

chore(doc): fix typos in FDSPv2 doc (#7104)

tengomucho committed 1 year ago

Verified 7350b702

add example for flash attnetion (#7098)

JackCaoG committed 1 year ago

Verified 8ae4c769

[Pallas] Refactor the gmm kernel (#7099)

alanwaketan committed 1 year ago

Verified 5327033b

reorganize the example dir (#7097)

JackCaoG committed 1 year ago

Verified c294625d

Add examples for how to benchmark a PyTorch/XLA model (#7089)

JackCaoG committed 1 year ago

Verified 0ce06eca

Update spmd doc (#7096)

JackCaoG committed 1 year ago

Verified 8a1ada88

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

JackCaoG committed 1 year ago

Verified f336317e

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

GleasonK committed 1 year ago

Verified baf08aea

Add dlpack support (#7025)

vanbasten23 committed 1 year ago

Verified 60238557

add missing aten op (#7078)

zpcore committed 1 year ago

Verified 5e1d454e

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

ysiraichi committed 1 year ago

Verified 8d35eb05

Dump HLO HBM usage info (#7085)

JackCaoG committed 1 year ago

Verified 206f1b7f

Implement `ComputationClient::GetMemoryInfo` (#7086)

will-cromar committed 1 year ago

Verified 5409cd5b

add example for fsdp (#7061)

JackCaoG committed 1 year ago

Verified 961c22ae

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

Older