Code

Support tuples in partition spec#5488

Code

Merged

jonb377 merged 9 commits intomasterpytorch/xla:masterfrom jonbolin-tuplepytorch/xla:jonbolin-tupleCopy head branch name to clipboardAug 25, 2023

+148-32

Commits

Commits on Aug 25, 2023

• Support tuples in partition spec

  jonb377

  committed
  4bb16b9

  Copy full SHA for 4bb16b9

• Add unit test for partial replication

  jonb377

  committed
  9aa32d6

  Copy full SHA for 9aa32d6

• yapf

  jonb377

  committed
  17416da

  Copy full SHA for 17416da

• Support higher-rank tensors over lower-rank mesh

  jonb377

  committed
  6e19717

  Copy full SHA for 6e19717

• Fix test & yapf

  jonb377

  committed
  8493547

  Copy full SHA for 8493547

• Don't use partition_spec when creating group assignment

  jonb377

  committed
  2ac4dd4

  Copy full SHA for 2ac4dd4

• Update documentation

  jonb377

  committed
  b7f5a61

  Copy full SHA for b7f5a61

• More documentation

  jonb377

  committed
  54258af

  Copy full SHA for 54258af

• Translate named specs in ShardingSpec

  jonb377

  committed
  1006a89

  Copy full SHA for 1006a89