pytorch/xla

Pull Requests Commits

Remove SPMD stale files in experimental

JackCaoG committed 1 year ago

479b78cd

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

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

qihqi committed 1 year ago

Verified 32595312

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

bhavya01 committed 1 year ago

Verified f448b066

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

ManfeiBai committed 1 year ago

Verified 6c4d3d84

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

vanbasten23 committed 1 year ago

Verified 6816ac8c

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

qihqi committed 1 year ago

Verified ea0fce2b

Initialize benchmarks with device type. (#7321)

ysiraichi committed 1 year ago

Verified bb27cb2a

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

huzama committed 1 year ago

Verified 85143432

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

JackCaoG committed 1 year ago

Verified 73923ecd

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

miladm committed 1 year ago

Verified 71578d72

only run eager test on v4 CI (#7315)

JackCaoG committed 1 year ago

Verified e85d292a

Type annotation for benchmarks/ (#7289)

zpcore committed 1 year ago

Verified 98dd99e7

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

JackCaoG committed 1 year ago

Verified cb6549aa

Update XLA Pin (#7308)

JackCaoG committed 1 year ago

Verified 2cfa13ca

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

qihqi committed 1 year ago

Verified c2753715

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

ManfeiBai committed 1 year ago

Verified 91389c7c

[benchmarks] Initialize `moco` and TPU experiments on XLA. (#7257)

ysiraichi committed 1 year ago

Verified abf5c31b

reduce dcoder model size to make eager runable on v2-8 (#7288)

JackCaoG committed 1 year ago

Verified a0b76fce

use the eager_mode_context for some of the tests (#7276)

JackCaoG committed 1 year ago

Verified 1492eba8

Add tpu device id for v6e (#7269)

bhavya01 committed 1 year ago

Verified e8601ea3

Use np.prod instead of np.product (#7301)

JackCaoG committed 1 year ago

Verified 5fa23c1a

Older