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

Merged

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

JackCaoG merged 5 commits into master from JackCaoG/ir_value_sharding

sharding should be per output of IR Node, instead of per IR Node

1df2c171

JackCaoG added distributed

Update sharding_hash method

7ee743c5

Add test for sharding on IR with multiple output

ef5df5f1

fix cpu test

f1fd96d5

JackCaoG changed the title ~~[WIP]Sharding should be per output of IR Node, instead of per IR Node~~ Sharding should be per output of IR Node, instead of per IR Node 2 years ago

JackCaoG requested a review from

alanwaketan 2 years ago

JackCaoG requested a review from

jonb377 2 years ago

jonb377 approved these changes on 2023-07-21

alanwaketan commented on 2023-07-21

alanwaketan approved these changes on 2023-07-21

Fix a bug in getSharding

0c533f31

JackCaoG merged 901d154a into master 2 years ago

Reviewers

alanwaketan

jonb377

Assignees

No one assigned

Labels

distributed

Milestone

No milestone

xla Sharding should be per output of IR Node, instead of per IR Node #5330 Merged

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

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

Merged