PR #7248 [SPMD] Fix reduce_scatter for TPUv2

[SPMD] Fix reduce_scatter for TPUv2 #7248

alanwaketan merged 6 commits into master from alanwaketan/fix2

alanwaketan added tpuci

alanwaketan requested a review from

jonb377 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

JackCaoG added backport_2.4

JackCaoG approved these changes on 2024-06-11

Fix tpu v3

4516f9dc

skip tpu v2

bb4f1796

Fix linters

3415c6d0

Fix

f92a4dd8

Add a debug print

658f68f0

alanwaketan force pushed from 746f416b to 658f68f0 1 year ago

remove print

6a15573d

alanwaketan merged 0025ca76 into master 1 year ago

Reviewers

JackCaoG

jonb377

Assignees

alanwaketan

Labels

backport_2.4

Milestone

No milestone