pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

e5dde4a6

Your commit message

alanwaketan committed 1 year ago

d41c4f44

alanwaketan committed 1 year ago

7e61cc3a

alanwaketan committed 1 year ago

7964f442

alanwaketan committed 1 year ago

ef591456

make gmm functional on xla

alanwaketan committed 1 year ago

85e77c4b

alanwaketan committed 1 year ago

86925d75

Make _make_group_metadata work for xla

alanwaketan committed 1 year ago

c754762e

alanwaketan committed 1 year ago

615ace3a

use repeat_with_fixed_output_size

alanwaketan committed 1 year ago

453017d7

[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

Older