Code

Add device map config#331

Code

Merged

mrwyattii merged 7 commits intomaindeepspeedai/DeepSpeed-MII:mainfrom mrwyattii/deploy-rank-refactordeepspeedai/DeepSpeed-MII:mrwyattii/deploy-rank-refactorCopy head branch name to clipboardDec 15, 2023

+185-41

Commits

Commits on Nov 27, 2023

• add deploy rank config

  mrwyattii

  committed
  c4a056c

  Copy full SHA for c4a056c

Commits on Nov 28, 2023

• refactor deploy_rank to device_map

  mrwyattii

  committed
  24fbf9d

  Copy full SHA for 24fbf9d

• remove prototyping code

  mrwyattii

  committed
  cdb3635

  Copy full SHA for cdb3635

• remove deploy_rank

  mrwyattii

  committed
  7fddcc9

  Copy full SHA for 7fddcc9

Commits on Dec 15, 2023

• assert that length of device map must match replica_num

  mrwyattii

  committed
  eca71c2

  Copy full SHA for eca71c2

• add unit test

  mrwyattii

  committed
  4884a1e

  Copy full SHA for 4884a1e

• Merge branch 'main' into mrwyattii/deploy-rank-refactor

  mrwyattii

  authored
  c012f9d

  Copy full SHA for c012f9d