pytorch/xla

Pull Requests Commits

[Op info test] Implemented the following ops: select_scatter and signbit (#7892)

yenkwang committed 1 year ago

Verified 5755d77d

DYNAMO RNG seed update optimization (#7884)

JackCaoG committed 1 year ago

Verified 3776cccc

Reenable test_traceable_collectives.py (#7881)

JackCaoG committed 1 year ago

Verified d284a21e

Fix few issues found during copybara import to Google3. (#7880)

qihqi committed 1 year ago

Verified f8905a32

fix wheel name for pip==24.2 (#7883)

zpcore committed 1 year ago

Verified defe6151

[torch_xla2] Add support for torch.qr and torch.linalg.qr (#7885)

danielvegamyhre committed 1 year ago

Verified b162b66d

update wheel name (#7882)

zpcore committed 1 year ago

Verified e30927a5

[torch_xla2] Fix all aten jax ops getter (#7868)

chunnienc committed 1 year ago

Verified b075cf81

fix benchamrk batch_size report (#7877)

zpcore committed 1 year ago

Verified 940a4b1b

[torchxla2] enable exp to cfloat (#7861)

ManfeiBai committed 1 year ago

Verified bda425fa

Fix `norm` data-type when using AMP. (#7878)

ysiraichi committed 1 year ago

Verified ac13bf23

[torch_xla2] Add randint_like (#7879)

danielvegamyhre committed 1 year ago

Verified 51b8b70f

[torch_xla2] Support operator `ones_like`, `zeros_like`, `full_like` (#7875)

guyao committed 1 year ago

Verified bbe19926

[torch_xla2] Enable test_ops frac..hstack (#7871)

gmittert committed 1 year ago

Verified 51704c6a

update nightly wheel naming (#7873)

zpcore committed 1 year ago

Verified ac7fd44f

As_stride should aplly on input tensor, not the base tensor (#7864)

JackCaoG committed 1 year ago

Verified 0e35022f

Add Python binding for xla::DotGeneral (#7863)

lsy323 committed 1 year ago

Verified 37312c19

Enable ops flatten, flip. fliplr, flipud, fmod (#7869)

gmittert committed 1 year ago

Verified e9b2f022

[torchxla2] Fix README venv setup (#7870)

gmittert committed 1 year ago

Verified ab3e4e56

type hint for xla_model (#7836)

zpcore committed 1 year ago

Verified fb29a49e

disable test_traceable_collectives (#7866)

JackCaoG committed 1 year ago

Verified 3f4fdf0e

[torch_xla2] Fix aten.copy (#7830)

chunnienc committed 1 year ago

Verified 3cd2b52d

[torch_xla2] Seperate pooled and indices tensor computations for max_pool2d_with_indices (#7829)

chunnienc committed 1 year ago

Verified 52571833

[torch_xla2] Cast index tensor dtype in aten.index_select (#7828)

chunnienc committed 1 year ago

Verified 4b27b3f1

[torch_xla2] DistributedDataParallel wrapper with minGPT example (#7847)

will-cromar committed 1 year ago

Verified b9b5653e

lower ops for issue #7557 (#7835)

wenxindongwork committed 1 year ago

Verified b8db9322

Add fp8e4m3fn support (#7842)

lsy323 committed 1 year ago

Verified 1fb09bc6

[torch_xla2] Setup nightly build config and script (#7856)

chunnienc committed 1 year ago

Verified 41bf6dab

[Pallas] Support ab for flash_attention (#7840)

alanwaketan committed 1 year ago

Verified 21a0b5a7

update wheel name (#7722)

zpcore committed 1 year ago

Verified c6ba0e98

Newer Older