Code

Support torch async dist checkpoint#2612

Code

Draft

novahow wants to merge 9 commits intoflyteorg:masterflyteorg/flytekit:masterfrom novahow:plugins/torch_asyncnovahow/flytekit:plugins/torch_asyncCopy head branch name to clipboard

+64-4

Commits

Commits on Dec 13, 2024

• torchCkpt

  Show description for 0a08cdf
  novahow

  committed
  0a08cdf

• add torchckpt to synccheckpoint

  Show description for 23530dc
  novahow

  committed
  23530dc

• add torchckpt, need refactor

  Show description for d1c3dd1
  novahow

  committed
  d1c3dd1

• add torch dep back and simplify torchSyncCkpt

  Show description for 010597d
  novahow

  committed
  010597d

• remove redundant resotre func

  Show description for 5b7a2ec
  novahow

  committed
  5b7a2ec

• fix rebase issue

  Show description for 05c0c48
  novahow

  committed
  05c0c48

• remove existing torchAsync instance

  Show description for a3fafa1
  novahow

  committed
  a3fafa1

• exp with union imagespec

  Show description for d3fc20d
  novahow

  committed
  d3fc20d

• fix ihheritance

  Show description for 340365e
  novahow

  committed
  340365e