pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

c5253ce8

add a test case

alanwaketan committed 2 years ago

c1c9f8bc

alanwaketan committed 2 years ago

11072ef3

[SPMD] Expose apply_backward_optimization_barrier (#7477)

alanwaketan committed 2 years ago

Verified 6894a08c

Remove SPMD stale files in experimental (#7499)

JackCaoG committed 2 years ago

Verified d028176a

fix parameter sharding in testing script (#7496)

JackCaoG committed 2 years ago

Verified 4caead75

Bump version to r2.5 (#7344)

will-cromar committed 2 years ago

Verified e2868f10

Add bitwise_left_shift, bitwise_right_shift (#7343)

ManfeiBai committed 2 years ago

Verified 557996a6

Document the CUDA plugin in README (#7345)

will-cromar committed 2 years ago

Verified b5052880

Register sdp lower precision autocast (#7299)

lingzhi98 committed 2 years ago

Verified 53c77e2e

Handle random seed for eager mode (#7328)

JackCaoG committed 2 years ago

Verified f570830a

torch_xla.manual_seed (#7340)

JackCaoG committed 2 years ago

Verified e294bb22

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

JackCaoG committed 2 years ago

Verified 222bbd8f

Support cumprod (#7339)

lsy323 committed 2 years ago

Verified dcbd9296

Handle non persistent buffer in to_xla (#7338)

lsy323 committed 2 years ago

Verified 0a97d771

Modify op info test to check dtypes. Fix up few dtypes. (#7323)

qihqi committed 2 years ago

Verified 32595312

Update 2.4 rc-6 in build trigger (#7335)

bhavya01 committed 2 years ago

Verified f448b066

Add support for baddbmm, bincount, tan, tanh (#7312)

ManfeiBai committed 2 years ago

Verified 6c4d3d84

Avoid OOM when build PyTorch with CUDA enabled. (#7329)

vanbasten23 committed 2 years ago

Verified 6816ac8c

Make conversion method like to_xla works inside of mode (#7319)

qihqi committed 2 years ago

Verified ea0fce2b

Initialize benchmarks with device type. (#7321)

ysiraichi committed 2 years ago

Verified bb27cb2a

Fix: Ensure checkpoint deletion only occurs on rank 0 (#7296)

huzama committed 2 years ago

Verified 85143432

Egaerly execute inplace ops if in eager mode (#7322)

JackCaoG committed 2 years ago

Verified 73923ecd

Add support for trig, binary, algebraic ops (#7325)

miladm committed 2 years ago

Verified 71578d72

only run eager test on v4 CI (#7315)

JackCaoG committed 2 years ago

Verified e85d292a

Type annotation for benchmarks/ (#7289)

zpcore committed 2 years ago

Verified 98dd99e7

Update the instruction for how to pin upstream pr (#7314)

JackCaoG committed 2 years ago

Verified cb6549aa

Update XLA Pin (#7308)

JackCaoG committed 2 years ago

Verified 2cfa13ca

Wrap adding logging options in try / catch. (#7307)

qihqi committed 2 years ago

Verified c2753715

[Fori_loop|While_loop] Enable while_loop/fori_loop, add test case (#7157)

ManfeiBai committed 2 years ago

Verified 91389c7c

Older