pytorch/xla

Pull Requests Commits

ManfeiBai committed 1 year ago

Verified 7fb69b8d

ManfeiBai committed 1 year ago

Verified 877b30f4

try xla::status

ManfeiBai committed 1 year ago

Verified 7a2697e2

try xla::status

ManfeiBai committed 1 year ago

Verified 9c2bd8a4

try xla::status

ManfeiBai committed 1 year ago

Verified e8b85116

try xla::status

ManfeiBai committed 1 year ago

Verified 821caf04

try xla::status

ManfeiBai committed 1 year ago

Verified ab3a1eb4

try xla::status

ManfeiBai committed 1 year ago

Verified 96b879ce

try xla::status

ManfeiBai committed 1 year ago

Verified 089638a2

try xla::status

ManfeiBai committed 1 year ago

Verified 7122b5cb

try xla::status

ManfeiBai committed 1 year ago

Verified 1ce954e1

Merge branch 'master' into ManfeiBai-patch-75

ManfeiBai committed 1 year ago

Verified 53ee7b5c

[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

ManfeiBai committed 1 year ago

Verified b95a83ff

Update ifrt_computation_client_test.cc

ManfeiBai committed 1 year ago

Verified 334a09f2

Update pjrt_computation_client_test.cc

ManfeiBai committed 1 year ago

Verified a8becc81

Update pjrt_registry.cc

ManfeiBai committed 1 year ago

Verified 0cee9cbf

Update pjrt_computation_client.cc

ManfeiBai committed 1 year ago

Verified 2cfaaff2

Update init_python_bindings.cpp

ManfeiBai committed 1 year ago

Verified e22257fb

Update convolution_helper.h

ManfeiBai committed 1 year ago

Verified b538424d

Update convolution_helper.cpp

ManfeiBai committed 1 year ago

Verified 9fd2f9d1

Update convolution_helper.h

ManfeiBai committed 1 year ago

Verified 7e5f76fd

[Pallas] Support Dynamo (#6477)

alanwaketan committed 1 year ago

Verified ce8ee38e

Older