pytorch/xla

Pull Requests Commits

remove torch_xla2 jax dependency in native install (#7126)

zpcore committed 1 year ago

Verified ed90be1e

Use remote cache for `push` events too (#7124)

will-cromar committed 1 year ago

Verified 7770a494

`index`: fix index of 0-element tensor by 0-element tensor. (#7113)

ysiraichi committed 1 year ago

Verified be3b08e6

[Pallas] Set a better tiling for gmm (#7119)

alanwaketan committed 1 year ago

Verified fd4900ce

[Pallas] Make gmm output a tensor (#7120)

alanwaketan committed 1 year ago

Verified 65b5ace8

Only use remote cache for main repository, not forks (#7112)

will-cromar committed 1 year ago

Verified 1a8c2fe2

[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

Newer Older