pytorch/xla

Pull Requests Commits

update load ipynb for issue_7331_test_code

ManfeiBai committed 1 year ago

Verified 4a002975

Update pallas doc with paged_attention (#7591)

wonjoo-wj committed 1 year ago

Verified 3bdca4d1

should_donate should not propagate to the output data (#7585)

JackCaoG committed 1 year ago

Verified 6410989c

Add documentation on fixing a torchbench model on torchxla2 (#7583)

qihqi committed 1 year ago

Verified 6d375f2c

Use upstream trilinear decomp by default (#7586)

JackCaoG committed 1 year ago

Verified 5e51ca23

Add an example of using GKE with torchrun (#7589)

will-cromar committed 1 year ago

Verified 77f69434

Remove references to old release project (#7587)

will-cromar committed 1 year ago

Verified d66a67b5

remove XLA_USE_BF16 and other variants (#7582)

JackCaoG committed 1 year ago

Verified 7d41035b

Clean up experimental (#7579)

JackCaoG committed 1 year ago

Verified ed15ea1e

Add `jlibrary` module to torch_xla2 for preserving library abstractions during export (#7128)

GleasonK committed 1 year ago

Verified 91d462cc

[torch_xla2] Prototype of `torch.distributed` backend (#7311)

will-cromar committed 1 year ago

Verified 447031e6

Enable eager spmd (#7341)

JackCaoG committed 1 year ago

Verified d5e57135

Add support of "nn.functional.pdist" (#7578)

ManfeiBai committed 1 year ago

Verified c0942937

Add helper function to get 1d mesh (#7577)

JackCaoG committed 1 year ago

Verified ab41f787

tests for tpu-info CLI (#7337)

will-cromar committed 1 year ago

Verified 81829941

[SPMD] Support manual all-reduce (#7576)

alanwaketan committed 1 year ago

Verified 0df5c291

move dynamo_mark_sharding and dynamo_set_buffer_donor_ out of experim… (#7502)

JackCaoG committed 1 year ago

Verified 016f7cc2

Fix spmd reduce scatter python test (#7500)

bhavya01 committed 1 year ago

Verified c654f12f

[SPMD] Expose apply_backward_optimization_barrier (#7477)

alanwaketan committed 1 year ago

Verified 6894a08c

Remove SPMD stale files in experimental (#7499)

JackCaoG committed 1 year ago

Verified d028176a

fix parameter sharding in testing script (#7496)

JackCaoG committed 1 year ago

Verified 4caead75

Bump version to r2.5 (#7344)

will-cromar committed 1 year ago

Verified e2868f10

Add bitwise_left_shift, bitwise_right_shift (#7343)

ManfeiBai committed 1 year ago

Verified 557996a6

Document the CUDA plugin in README (#7345)

will-cromar committed 1 year ago

Verified b5052880

Register sdp lower precision autocast (#7299)

lingzhi98 committed 1 year ago

Verified 53c77e2e

Handle random seed for eager mode (#7328)

JackCaoG committed 1 year ago

Verified f570830a

torch_xla.manual_seed (#7340)

JackCaoG committed 1 year ago

Verified e294bb22

Support eager mode for multi-process training (#7327)

JackCaoG committed 1 year ago

Verified 222bbd8f

Support cumprod (#7339)

lsy323 committed 1 year ago

Verified dcbd9296

Handle non persistent buffer in to_xla (#7338)

lsy323 committed 1 year ago

Verified 0a97d771

Older