pytorch/xla

Pull Requests Commits

Clean up test infra and rebase on master

changm committed 1 year ago

Verified 4a5ee054

Clean up printfs

changm committed 1 year ago

Verified 67c13710

Prototype bucketing in dynamo XLA bridge

changm committed 1 year ago

Verified 3da601b5

[Pallas] Support programmatically extracting the payload (#6696)

alanwaketan committed 1 year ago

Verified 37067917

Ops refactor 1 (#6671)

qihqi committed 1 year ago

Verified 60431850

Revert "Removing nondeterminism from partitioner eliminates one graph" (#6703)

JackCaoG committed 1 year ago

Verified 707a6329

Make use_spmd to force SPMD mode for easier 3rd-party integration. (#6680)

yeounoh committed 1 year ago

Verified f3a2e243

[Fori_loop|While_loop] Enable fori_loop with add/sub test case (#6603)

ManfeiBai committed 1 year ago

Verified 6170df58

Revert "Change ATEN generator argument type to const std::optional<Generator>&" (#6701)

alanwaketan committed 1 year ago

Verified 6cc5c381

Not building cuda plugins to unblock upstream ci. (#6699)

vanbasten23 committed 1 year ago

Verified e99ab496

Removing nondeterminism from partitioner eliminates one graph (#6675)

ezyang committed 1 year ago

Verified d28053f9

Run test gpu device detection without setting PJRT_DEVICE (#6650)

changm committed 1 year ago

Verified 57f4780d

[Infra] Bump python version to 3.10 (#6685)

alanwaketan committed 1 year ago

Verified 99c41bae

Revert "Make `as_strided_copy` materialize a new tensor with `index`.… (#6693)

lsy323 committed 1 year ago

Verified 1fd1afb6

[OpenXLA|TSL] Replace tsl::Status to absl::Status in profiler.h/.cc (#6655)

ManfeiBai committed 1 year ago

Verified 13dc9df4

[OpenXLA|TSL] Replace ::tsl::gtl::InlinedVector with ::absl::InlinedVector convolution_helper.h (#6656)

ManfeiBai committed 1 year ago

Verified d2090624

[SPMD][DTensor] introduce xla_distribute_module for DTensor integration (#6683)

yeounoh committed 1 year ago

Verified b6b9c6da

Add SPMD on GPU instructions (#6684)

vanbasten23 committed 1 year ago

Verified cdd6466c

[DOC] Update the camponion PR instructions (#6687)

alanwaketan committed 1 year ago

Verified 39db1381

Change ATEN generator argument type to const std::optional<Generator>& (#6686)

alanwaketan committed 1 year ago

Verified 8078b8f3

[SPMD] Check the real physical device type if XlaDeviceType::SPMD (#6652)

yeounoh committed 1 year ago

Verified 46e22300

[OpenXLA|TSL] Replace `tsl/platform/stringpiece.h` with `absl/strings/string_view.h` (#6657)

ManfeiBai committed 1 year ago

Verified 9c5c6d91

Revert "Add missing toml package to post-submit CI" (#6682)

vanbasten23 committed 1 year ago

Verified 1dc78948

Add missing shell options to `test/tpu/run_tests.sh` (#6681)

will-cromar committed 1 year ago

Verified 3dbcbaab

Add Dockerfile for TPU CI self-hosted runner (#6678)

mbzomowski committed 1 year ago

Verified 6bcf4fd3

nightly.sh: do not include one-off jobs in generated reports (#6679)

cota committed 1 year ago

Verified 6fe19d3c

[Pallas] Support Flash Attention (#6658)

alanwaketan committed 1 year ago

Verified 9d4dcae7

Remove `kubectl` install from TPU CI script (#6672)

will-cromar committed 1 year ago

Verified 57b07c3e

[Pallas] Support Dynamo (#6477)

alanwaketan committed 1 year ago

Verified ce8ee38e

Make `as_strided_copy` materialize a new tensor with `index`. (#6624)

ysiraichi committed 1 year ago

Verified 3abc21df

Older