pytorch/xla

Pull Requests Commits

Update openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified acdec664

Update openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified de67ffa8

Update openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified d03a9505

Update openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified d92b8c02

Update openxla_pin_update_weekly.yml with create new branch for each PR

ManfeiBai committed 1 year ago

Verified ebdb09ff

Update openxla_pin_update_weekly.yml to add tpuci label

ManfeiBai committed 1 year ago

Verified 88ac75ed

Update openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified 0b29eb71

add workflow_dispatch to enable manual trigger

ManfeiBai committed 1 year ago

Verified d3b1717c

Rename _openxla_pin_update_weekly.yml to openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified 265d2cb9

Update _openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified ab6a1bc0

ManfeiBai committed 1 year ago

Verified 00384819

Create _openxla_pin_update_weekly.yml

ManfeiBai committed 1 year ago

Verified 2fb1c878

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

Older