Code

[SPMD] Support mark_sharding on IRs#5301

Code

Merged

alanwaketan merged 5 commits intomasterpytorch/xla:masterfrom alanwaketan/spmd_irpytorch/xla:alanwaketan/spmd_irCopy head branch name to clipboardJul 13, 2023

+28-1

Commits

Commits on Jul 13, 2023

• [SPMD] Avoid recompilations in xs.mark_sharding()

  Show description for eb15142
  alanwaketan

  committed
  eb15142

  Copy full SHA for eb15142

• [SPMD] Support mark_sharding on IRs

  Show description for 938d5fb
  alanwaketan

  committed
  938d5fb

  Copy full SHA for 938d5fb

• Fix rebase error

  alanwaketan

  committed
  632c9fe

  Copy full SHA for 632c9fe

• Fix linters

  alanwaketan

  committed
  d22e642

  Copy full SHA for d22e642

• Unskip test_2d_tensor_3d_mesh

  alanwaketan

  committed
  bab326e

  Copy full SHA for bab326e