Code

Model partitioning#28

Code

Merged

xrsrke merged 22 commits intoxrsrke:mainxrsrke/pipegoose:mainfrom abourramouss:mainabourramouss/pipegoose:mainCopy head branch name to clipboardNov 22, 2023

+255-58

Commits

Commits on Nov 4, 2023

• first approach using list flattener.

  abourramouss

  committed
  53fa580

  Copy full SHA for 53fa580

• remove print from split method

  abourramouss

  committed
  afc7c30

  Copy full SHA for afc7c30

Commits on Nov 15, 2023

• use MPTracer (HFTracer)

  abourramouss

  committed
  f431b10

  Copy full SHA for f431b10

Commits on Nov 17, 2023

• copy from fairscale

  abourramouss

  committed
  48a099b

  Copy full SHA for 48a099b

• remove unused leaf tracer

  abourramouss

  committed
  89e473a

  Copy full SHA for 89e473a

Commits on Nov 18, 2023

• new split nodes logic

  abourramouss

  committed
  1d0c0f1

  Copy full SHA for 1d0c0f1

• partitioning WIP

  danielgrittner

  committed
  50bdf0c

  Copy full SHA for 50bdf0c

• Merge branch 'main' into partitioning-impl

  abourramouss

  authored
  9644b5d

  Copy full SHA for 9644b5d

• Merge pull request #1 from abourramouss/partitioning-impl

  Show description for 1ea8f0b
  abourramouss

  authored
  1ea8f0b

  Copy full SHA for 1ea8f0b

Commits on Nov 19, 2023

• Working partitioning implementation

  danielgrittner

  committed
  f36b085

  Copy full SHA for f36b085

• Merge branch 'main' into working-partitioning-impl

  abourramouss

  authored
  2e837a7

  Copy full SHA for 2e837a7

• Merge pull request #2 from abourramouss/working-partitioning-impl

  Show description for a710971
  abourramouss

  authored
  a710971

  Copy full SHA for a710971

• clean up model partitioning code

  danielgrittner

  committed
  2e06cdb

  Copy full SHA for 2e06cdb

• Merge pull request #3 from abourramouss/partitioning-cleanup

  Show description for 35a912f
  abourramouss

  authored
  35a912f

  Copy full SHA for 35a912f

• fix partitioning

  danielgrittner

  committed
  290ae68

  Copy full SHA for 290ae68

• remove unused function

  danielgrittner

  committed
  d7c3cf2

  Copy full SHA for d7c3cf2

• ability to test multiple models

  abourramouss

  committed
  4494d62

  Copy full SHA for 4494d62

• Update test_partitioner.py

  abourramouss

  committed
  fa36cab

  Copy full SHA for fa36cab

Commits on Nov 20, 2023

• Update test_partitioner.py

  abourramouss

  committed
  4c86ef9

  Copy full SHA for 4c86ef9

Commits on Nov 21, 2023

• fix model partitioning and add more tests

  danielgrittner

  committed
  e69466d

  Copy full SHA for e69466d

• Merge pull request #4 from abourramouss/fix-model-partitioning

  Show description for 37f112b
  abourramouss

  authored
  37f112b

  Copy full SHA for 37f112b

Commits on Nov 22, 2023

• Merge branch 'xrsrke:main' into main

  abourramouss

  authored
  589f5f7

  Copy full SHA for 589f5f7