Code

IFRT prototype#5677

Code

Merged

will-cromar merged 33 commits intomasterpytorch/xla:masterfrom wcromar/ifrt-prototypepytorch/xla:wcromar/ifrt-prototypeCopy head branch name to clipboardDec 15, 2023

+1,223-146

Commits

Commits on Dec 14, 2023

• Start IFRT prototype

  will-cromar

  committed
  fdd13af

  Copy full SHA for fdd13af

• remove comment

  will-cromar

  committed
  c7e48b4

  Copy full SHA for c7e48b4

• formatting

  will-cromar

  committed
  1312465

  Copy full SHA for 1312465

• basic sharding

  will-cromar

  committed
  a7ae062

  Copy full SHA for a7ae062

• Add xla::OpSharding back as source of truth

  will-cromar

  committed
  835487f

  Copy full SHA for 835487f

• wrapping and unwrapping sharded data

  will-cromar

  committed
  a821a4d

  Copy full SHA for a821a4d

• ExecuteReplicated

  will-cromar

  committed
  2e8842b

  Copy full SHA for 2e8842b

• [revert later] try resharding

  will-cromar

  committed
  d75c040

  Copy full SHA for d75c040

• Revert "[revert later] try resharding"

  Show description for 5544fc0
  will-cromar

  committed
  5544fc0

  Copy full SHA for 5544fc0

• reassemble sharded outputs

  will-cromar

  committed
  03481e0

  Copy full SHA for 03481e0

• fix output devices

  will-cromar

  committed
  f2f8334

  Copy full SHA for f2f8334

• cleanup

  will-cromar

  committed
  c16ea62

  Copy full SHA for c16ea62

• fix rebase issues

  will-cromar

  committed
  4d8418a

  Copy full SHA for 4d8418a

• fix const plumbing

  will-cromar

  committed
  e5a13a2

  Copy full SHA for e5a13a2

• shared_ptr to unique_ptr

  will-cromar

  committed
  155a811

  Copy full SHA for 155a811

• parallelize input/output handling

  will-cromar

  committed
  b1238da

  Copy full SHA for b1238da

• fix concurrency issues

  will-cromar

  committed
  472b9d4

  Copy full SHA for 472b9d4

• remove some commented out code

  will-cromar

  committed
  a6544b7

  Copy full SHA for a6544b7

• formatting

  will-cromar

  committed
  5f6b4af

  Copy full SHA for 5f6b4af

• for coordinator init

  will-cromar

  committed
  c0e1ce8

  Copy full SHA for c0e1ce8

• remove extra std::moves

  will-cromar

  committed
  1812afa

  Copy full SHA for 1812afa

• unit test

  will-cromar

  committed
  135e68d

  Copy full SHA for 135e68d

• tune parallelfors

  will-cromar

  committed
  8fce005

  Copy full SHA for 8fce005

• pjrt -> ifrt

  will-cromar

  committed
  86883ad

  Copy full SHA for 86883ad

• fix rebasing issues

  will-cromar

  committed
  07a9efb

  Copy full SHA for 07a9efb

• formatting

  will-cromar

  committed
  3bfaa3a

  Copy full SHA for 3bfaa3a

• fix timer and comp env hash

  will-cromar

  committed
  a71a9d5

  Copy full SHA for a71a9d5

• formatting

  will-cromar

  committed
  0d2c842

  Copy full SHA for 0d2c842

• remove dead code

  will-cromar

  committed
  5b26aa4

  Copy full SHA for 5b26aa4

• Move SPMD string constant

  will-cromar

  committed
  0df165a

  Copy full SHA for 0df165a

• fix compile error

  will-cromar

  committed
  723191f

  Copy full SHA for 723191f

• format

  will-cromar

  committed
  9ebc616

  Copy full SHA for 9ebc616

• remove SPMD from hash

  will-cromar

  committed
  cc4c93d

  Copy full SHA for cc4c93d