pytorch/xla

Pull Requests Commits

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified fdaa2711

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified cf790501

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified 195e10ca

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified a72f6b01

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified 9d882917

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified b291b5d0

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified 746a67cb

Update test_dynamo.py

ManfeiBai committed 2 years ago

Verified b6fcee93

[BE] use self.assertEquals instead of str equality in test_zero1.py (#5364)

janeyx99 committed 2 years ago

Verified 8ada3332

Check the actual device instead of query env var for virtual device (#5352)

JackCaoG committed 2 years ago

Verified d09cc690

Add python test for SPMD+Runtime Python API (#5349)

JackCaoG committed 2 years ago

Verified e70fb298

fix amp dtype setting for GPU (#5337)

baoleai committed 2 years ago

Verified 22ba4af2

Fix CPU/GPU test coverage (#5355)

lsy323 committed 2 years ago

Verified 443fad11

remove unused patches (#5353)

lsy323 committed 2 years ago

Verified 1fa34bf5

Add py3.10 whl path to doc, refactor whl table (#5354)

lsy323 committed 2 years ago

Verified b95554b6

add fs linker flag (#5347)

lsy323 committed 2 years ago

Verified 5673ac4a

Make all-reduce a no-op when world size is 1 (#5342)

will-cromar committed 2 years ago

Verified 61816b0b

Also dump output sharding on HLO file (#5339)

JackCaoG committed 2 years ago

Verified efb1e6fd

Check out the release branch instead of origin/master in ansible (#5344)

vanbasten23 committed 2 years ago

Verified 68d7fe80

OpenXLA pin update 07052023 (#5304)

lsy323 committed 2 years ago

Verified 8ffbd1a4

Update Python device API for SPMD (#5129)

JackCaoG committed 2 years ago

Verified f996701a

Sharding should be per output of IR Node, instead of per IR Node (#5330)

JackCaoG committed 2 years ago

Verified 901d154a

handle single tensor for method send_to_device_single (#5317)

JackCaoG committed 2 years ago

Verified f5edcb20

Use `_sharded_cpu_state_dict` functionality to Write Items for SPMD Save Planner (#5315)

shahyash10 committed 2 years ago

Verified 74d115ec

Enable Some input output aliasing under SPMD (#5320)

JackCaoG committed 2 years ago

Verified ff8f7a49

Disable coverage for now (#5321)

JackCaoG committed 2 years ago

Verified e45cf404

Add tf independent serialization (#5308)

qihqi committed 2 years ago

Verified 0b5b5197

quick refactor on _get_group_assignment (#5318)

JackCaoG committed 2 years ago

Verified a8f723b6

Fix mismatched GPU docker image in the doc. (#5319)

vanbasten23 committed 2 years ago

Verified 42ae0bf9

Support unordered sharding spec for partial replication (#5316)

JackCaoG committed 2 years ago

Verified 78c6840c

Older