pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

fe6f9f21

Address comments and add tests

alanwaketan committed 2 years ago

ab44d7a9

alanwaketan committed 2 years ago

3ac7e63a

alanwaketan committed 2 years ago

b1d66f8b

Add Initial PoC and test

alanwaketan committed 2 years ago

006c48d5

[SPMD] Allow mark_shard scalar tensors (#6101)

alanwaketan committed 2 years ago

Verified 45a9e0e9

Remove flaky test (#6116)

jonb377 committed 2 years ago

Verified dd50c364

Fix model initialization. (#6076)

ysiraichi committed 2 years ago

Verified 7786d5df

Handled lifted constants in torch export. (#6111)

qihqi committed 2 years ago

Verified d8d7e58b

[SPMD] Add apply_backward_optimization_barrier (#6097)

alanwaketan committed 2 years ago

Verified 07540f21

Fix bug for low-precision ranges (#5636)

jonb377 committed 2 years ago

Verified eb3c446f

Add Pattern Boundary Marking API (#5930)

lsy323 committed 2 years ago

Verified 8243b502

Add out-of-place all-gather coalesced (#6059)

jeffhataws committed 2 years ago

Verified e3012bf9

Revert prs to get heads green again (#6107)

JackCaoG committed 2 years ago

Verified 6ae66f78

Update benchmark readme (#6100)

frgossen committed 2 years ago

Verified 694047eb

Add out-of-place reduce-scatter coalescing (#6058)

jeffhataws committed 2 years ago

Verified ba7c3478

Add cuda bm dry run (#6093)

frgossen committed 2 years ago

Verified aa959af6

Add benchmark tests (#6082)

frgossen committed 2 years ago

Verified bbe1a8de

Add benchmark experiment test (#6081)

frgossen committed 2 years ago

Verified dbd3e33c

Allow scalar all gather (#5797)

bfontain committed 2 years ago

Verified b2243507

Increase delay in trace_detached test (#6075)

jonb377 committed 2 years ago

Verified 9be1e94c

Update Troubleshotting doc with new PT_XLA_DEBUG (#6039)

JackCaoG committed 2 years ago

Verified 45d4750b

Gather more summary data when running `result_analyzer.py`. (#6067)

ysiraichi committed 2 years ago

Verified 64239127

Move 0-dimensional tensors to CPU before copying to XLA. (#6071)

ysiraichi committed 2 years ago

Verified a80c1e7f

Reduce benchmark experiment filename string. (#6069)

ysiraichi committed 2 years ago

Verified 0857f2a0

Use `subprocess` for checking XLA supported devices. (#6068)

ysiraichi committed 2 years ago

Verified faae458d

Piz/release (#6049)

zpcore committed 2 years ago

Verified a923bfc6

ci: add benchmarks/run_tests.sh (#6043)

cota committed 2 years ago

Verified 0313c6ae

[OpenXLA-pin] Update OpenXLA-pin to Nov29 (#6053)

ManfeiBai committed 2 years ago

Verified 599b79ff

Move CUDA tensors to CPU before moving to XLA. (#6060)

ysiraichi committed 2 years ago

Verified d5df845b

Older