pytorch/xla

Pull Requests Commits

Initial pr to support dynamic shape detection

JackCaoG committed 2 years ago

f0f4088f

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

JackCaoG committed 2 years ago

Verified c26b19eb

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

will-cromar committed 2 years ago

Verified 52b81e41

remove openxla_eval in test (#7811)

zpcore committed 2 years ago

Verified c7d9755d

Support fp8e5m2 dtype (#7740)

lsy323 committed 2 years ago

Verified 1ed26267

[torch_xla2] Support `aten.avg_pool2d` and `aten.avg_pool3d` with `ceil_mode=True` (#7806)

chunnienc committed 2 years ago

Verified dd04c588

Allow user to pass a name to torch_xla.compile (#7802)

JackCaoG committed 2 years ago

Verified 6250c02d

Attach callback to a Tensor's underlying buffer (#7793)

will-cromar committed 2 years ago

Verified 02d83221

[benchmarks] Fix verifier error handling and OOM. (#7777)

ysiraichi committed 2 years ago

Verified 5bbc4b3d

Move `_xla_register_custom_call_target` implementation into `PjRtComputationClient` (#7801)

will-cromar committed 2 years ago

Verified 7135a7e9

[torch/xla2] Lower aten_copysign and corrcoef (#7713)

wenxindongwork committed 2 years ago

Verified a3dd4d81

Refresh xla2 doc (#7804)

qihqi committed 2 years ago

Verified 8969043e

Update XLA Pin (#7797)

lsy323 committed 2 years ago

Verified dd3b00ca

make StepMarker's wait block for async device execution (#7794)

JackCaoG committed 2 years ago

Verified dfd7b00d

Use reduce-scatter coalescing for FSDP (#6024)

jeffhataws committed 2 years ago

Verified 7fe070a6

Update function doc strings (#7757)

JackCaoG committed 2 years ago

Verified d29b7611

disable sdpa amp (#7800)

lsy323 committed 2 years ago

Verified 225ec7b3

Add 1-layer gradient accumulation test to check aliasing (#7692)

jeffhataws committed 2 years ago

Verified a5520e5d

Add .rst for quantized op doc (#7760)

lsy323 committed 2 years ago

Verified bc9b6061

Handle SPMD case inside of `ComputationClient::WaitDeviceOps` (#7796)

will-cromar committed 2 years ago

Verified 2766f9f7

Improve device auto-detection (#7787)

zpcore committed 2 years ago

Verified 2ba4e057

Using dummy store when user skips barrier to save open file descriptors (#6834)

barry-jin committed 2 years ago

Verified eef7bb4e

Use std::make_optional (#7795)

cyyever committed 2 years ago

Verified b9e75390

Implement torch.ops._c10d_functional.broadcast (#7770)

JackCaoG committed 2 years ago

Verified e4ca4956

Update the `PT_XLA_DEBUG` to make it recognize the `torch_xla.compile` (#7768)

JackCaoG committed 2 years ago

Verified 4e94ff67

fix comment in #7776 (#7789)

JackCaoG committed 2 years ago

Verified 52ea89ff

Add `+default` to libtpu package name (#7792)

will-cromar committed 2 years ago

Verified ff307735

Use launch() for all multiprocessing (#7764)

zpcore committed 2 years ago

Verified 6c18d50c

Support fullgraph for torch_xla.compile (#7776)

JackCaoG committed 2 years ago

Verified 731eb08b

[Pallas] Fix the doc (#7788)

alanwaketan committed 2 years ago

Verified 942ba7a6

Older