pytorch/xla

Pull Requests Commits

JackCaoG committed 1 year ago

d4860cd6

JackCaoG committed 1 year ago

3a0a324b

update examples

JackCaoG committed 1 year ago

892bbb73

Allow user to pass a name to torch_xla.compile

JackCaoG committed 1 year ago

53ac9ffe

Add .rst for quantized op doc (#7760)

lsy323 committed 1 year ago

Verified bc9b6061

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

will-cromar committed 1 year ago

Verified 2766f9f7

Improve device auto-detection (#7787)

zpcore committed 1 year ago

Verified 2ba4e057

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

barry-jin committed 1 year ago

Verified eef7bb4e

Use std::make_optional (#7795)

cyyever committed 1 year ago

Verified b9e75390

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

JackCaoG committed 1 year ago

Verified e4ca4956

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

JackCaoG committed 1 year ago

Verified 4e94ff67

fix comment in #7776 (#7789)

JackCaoG committed 1 year ago

Verified 52ea89ff

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

will-cromar committed 1 year ago

Verified ff307735

Use launch() for all multiprocessing (#7764)

zpcore committed 1 year ago

Verified 6c18d50c

Support fullgraph for torch_xla.compile (#7776)

JackCaoG committed 1 year ago

Verified 731eb08b

[Pallas] Fix the doc (#7788)

alanwaketan committed 1 year ago

Verified 942ba7a6

Update libtpu_storage_path (#7786)

bhuvanpkaruturi committed 1 year ago

Verified ef49ddd8

Update GPU Development Doc for Hermetic Python setup (#7781)

lsy323 committed 1 year ago

Verified af7d2ea6

Further reduce number of compiling procs (#7779)

lsy323 committed 1 year ago

Verified e52b53f1

reenable dynamo dynamic shape test (#7775)

JackCaoG committed 1 year ago

Verified b09c9699

Update eager doc to use torch_xla.compile (#7778)

JackCaoG committed 1 year ago

Verified 28de4eb3

Disable open PJRT for xal2 if OS env disable flag is true (#7769)

FanhaiLu1 committed 1 year ago

Verified fb2d4e14

disable dynamo dynamic shape test since it is broken by some upstream… (#7773)

JackCaoG committed 1 year ago

Verified 622f4f93

openxla pin update (#7758)

zpcore committed 1 year ago

Verified dc3b265a

clean up deprecated (#7753)

zpcore committed 1 year ago

Verified 5702dae1

Refactor torch_xla.experimental.compile and move it out of experimental (#7750)

JackCaoG committed 1 year ago

Verified bdd00e54

Make libtpu install more explict (#7737)

JackCaoG committed 1 year ago

Verified 009e31aa

Fix few ops (#7759)

qihqi committed 1 year ago

Verified 2b5538d6

Add a guide to set up Visual Studio Code with PyTorch/XLA (#7755)

will-cromar committed 1 year ago

Verified b2e5c7d2

Move fsdp autowrap test to v4 TPUCI only (#7751)

JackCaoG committed 1 year ago

Verified db7fada8

Older