pytorch/xla

Pull Requests Commits

Reenable test_traceable_collectives.py

JackCaoG committed 1 year ago

1497fc3a

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

fix batch_size insistent in report (#7820)

zpcore committed 1 year ago

Verified f0bbaebd

Fix input output alias for custom inplace ops (#7822)

yitongh committed 1 year ago

Verified 8f794884

Use local TPU log dir in CI (#7816)

will-cromar committed 1 year ago

Verified ae2a3a4b

Add user guide for generating stablehlo composite op (#7826)

lsy323 committed 1 year ago

Verified 934771e7

Update fixing_op_info_test.md with addition for decomposed ops (#7825)

ManfeiBai committed 1 year ago

Verified 2dcdd455

Upgrade torch dependency to 2.4.0 plus misc changes: (#7821)

qihqi committed 1 year ago

Verified 60b9dfe7

Add ops `nn.functional.celu ... nn.functional.silu` (#7818)

ManfeiBai committed 1 year ago

Verified ec994104

Use torch export in test_composite (#7819)

qihqi committed 1 year ago

Verified 2eb4a60e

[benchmarks] Guard empty metrics on verifier runs. (#7809)

ysiraichi committed 1 year ago

Verified 9fbd64ac

Revert "make StepMarker's wait block for async device execution" (#7815)

JackCaoG committed 1 year ago

Verified b4335f8e

Support mark_dynamic (#7812)

JackCaoG committed 1 year ago

Verified 08d9595c

torch::lazy::MakeNode -> torch_xla::MakeNode (#7807)

JackCaoG committed 1 year ago

Verified c26b19eb

Remove stale `cloud-tpu-client` dependency (#7814)

will-cromar committed 1 year ago

Verified 52b81e41

remove openxla_eval in test (#7811)

zpcore committed 1 year ago

Verified c7d9755d

Support fp8e5m2 dtype (#7740)

lsy323 committed 1 year ago

Verified 1ed26267

Older