xla
Cache OpSharding by mesh and partition_spec
#5593

Merged

Cache OpSharding by mesh and partition_spec #5593

jonb377 merged 5 commits into master from jonbolin-cache-ga

jonb377 requested a review from

yeounoh 2 years ago

jonb377 requested a review from

alanwaketan 2 years ago

jonb377 commented on 2023-09-15

alanwaketan approved these changes on 2023-09-15

JackCaoG added distributed

jonb377 force pushed from 8c6ea20a to 36025906 2 years ago

jonb377 changed the title ~~Cache group assignments~~ Cache OpSharding by mesh and partition_spec 2 years ago

jonb377 force pushed from 36025906 to e7116396 2 years ago

Cache OpSharding in mark_sharding

ee3e963b

jonb377 force pushed from e7116396 to ee3e963b 2 years ago

Cache global_runtime_device_count

a7b7cbcf

Add metric and unit test

9828c789

jonb377 force pushed from 62b989eb to 9828c789 2 years ago

alanwaketan approved these changes on 2023-09-18

Use functools.lru_cache in get_op_sharding

588eed18

yapf

3c2d0b4c

jonb377 merged 71999f59 into master 2 years ago

jonb377 deleted the jonbolin-cache-ga branch 2 years ago

yeounoh commented on 2023-09-21

Reviewers

alanwaketan

yeounoh

Assignees

No one assigned

Labels

distributed

Milestone

No milestone