Code

[SPMD] Hybrid Device mesh creation#5147

Code

Merged

khatwanimohit merged 14 commits intomasterpytorch/xla:masterfrom mohit/hybrid_meshpytorch/xla:mohit/hybrid_meshCopy head branch name to clipboardJun 19, 2023

+289-3

Commits

Commits on Jun 8, 2023

• hybrid_mesh class

  khatwanimohit

  committed
  b37d0e3

  Copy full SHA for b37d0e3

Commits on Jun 9, 2023

• hybrid_mesh class

  khatwanimohit

  committed
  3bd182c

  Copy full SHA for 3bd182c

• remove inherited fields

  khatwanimohit

  committed
  ad91169

  Copy full SHA for ad91169

Commits on Jun 10, 2023

• hybrid_mesh class

  khatwanimohit

  committed
  7b264ca

  Copy full SHA for 7b264ca

• lint fix

  khatwanimohit

  committed
  9f6d86c

  Copy full SHA for 9f6d86c

• Merge branch 'master' into mohit/hybrid_mesh

  khatwanimohit

  authored
  8f55df8

  Copy full SHA for 8f55df8

Commits on Jun 12, 2023

• fix test

  khatwanimohit

  committed
  c457f6c

  Copy full SHA for c457f6c

• fix lint

  khatwanimohit

  committed
  d71df3a

  Copy full SHA for d71df3a

Commits on Jun 13, 2023

• add unit test

  khatwanimohit

  committed
  ef665e9

  Copy full SHA for ef665e9

• skip test

  khatwanimohit

  committed
  9c6d8ab

  Copy full SHA for 9c6d8ab

• fix test

  khatwanimohit

  committed
  632cbbb

  Copy full SHA for 632cbbb

Commits on Jun 16, 2023

• add comments

  khatwanimohit

  committed
  572548b

  Copy full SHA for 572548b

• comments

  khatwanimohit

  committed
  640d0b3

  Copy full SHA for 640d0b3

• fix comment

  khatwanimohit

  committed
  abf04dc

  Copy full SHA for abf04dc