Skip to content

Add parallelism_auto flag to automatically set dp, tp and micro batch size - #516

Closed
pefontana wants to merge 28 commits into
mainfrom
hardcode-parallelism-data
Closed

Add parallelism_auto flag to automatically set dp, tp and micro batch size#516
pefontana wants to merge 28 commits into
mainfrom
hardcode-parallelism-data

Merge branch 'main' into hardcode-parallelism-data

e1f7276
Select commit
Loading
Failed to load commit list.
Garnix CI / devShell python [x86_64-linux] succeeded Mar 3, 2026 in 6m 42s

devShell python [x86_64-linux]

devShell python [x86_64-linux] succeeded

Details

Last 100 lines of logs:

these 7 derivations will be built:
  /nix/store/g5vws575wskdwwlky1ds0hgw23cdamhx-source.drv
  /nix/store/a6ms8lcw0ngv2xpc9wn2j3ddvcvc2wl3-psyche-workspace-deps-0.2.0.drv
  /nix/store/56h4wvkvpyk4c8flvqg5qx46pwb6jz1h-psyche-python-extension-0.2.0.drv
  /nix/store/c4bzilvbnmr0blbwbd72z4qf5g5pk3yp-install-agenix-shell.drv
  /nix/store/wzi8l8lnlg35h3dimy758k89dna3bdcd-psyche.drv
  /nix/store/sad59ziyza09k6ia7vzn0fpzhnbprl5i-psyche-python-env.drv
  /nix/store/fzvig7zaxm5g9mwgwn39rdrd2bd4mrqm-nix-shell.drv