Code

TRN2 Meshes and Configurations#916

Code

Merged

kelvin-zou merged 9 commits intoapple:mainapple/axlearn:mainfrom apoorvtintin:mainline-upstream-boilerplateapoorvtintin/axlearn:mainline-upstream-boilerplateCopy head branch name to clipboardFeb 13, 2025

+4,437-17

Commits

Commits on Feb 13, 2025

• TRN2 Meshes and Configurations

  apoorvtintin

  committed
  9e58b34

  Copy full SHA for 9e58b34

• Add get_recursive and set_recursive to ConfigBase.

  apoorvtintin

  committed
  a54dec0

  Copy full SHA for a54dec0

• Use loops inside get/set_recursively

  Show description for bb7fe11
  apoorvtintin

  committed
  bb7fe11

  Copy full SHA for bb7fe11

• Update partition spec

  apoorvtintin

  committed
  53f3622

  Copy full SHA for 53f3622

• Use get_recursively inside set

  apoorvtintin

  committed
  b8177ba

  Copy full SHA for b8177ba

• Move trn2 configs to a helper function.

  Show description for b0d2769
  apoorvtintin

  committed
  b0d2769

  Copy full SHA for b0d2769

• TRN2 partitionspec supports DP over FSDP and TP

  apoorvtintin

  committed
  ca00f37

  Copy full SHA for ca00f37

• Use for loop in get_recursively

  apoorvtintin

  committed
  25dd768

  Copy full SHA for 25dd768

• Update Golden Configs

  apoorvtintin

  committed
  2ee56ec

  Copy full SHA for 2ee56ec