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 dev-python [x86_64-linux] succeeded Mar 3, 2026 in 6m 38s

devShell dev-python [x86_64-linux]

devShell dev-python [x86_64-linux] succeeded

Details

Last 100 lines of logs:

these 6 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/wzi8l8lnlg35h3dimy758k89dna3bdcd-psyche.drv
  /nix/store/sad59ziyza09k6ia7vzn0fpzhnbprl5i-psyche-python-env.drv
  /nix/store/fzvig7zaxm5g9mwgwn39rdrd2bd4mrqm-nix-shell.drv
nix-shell> source: sourcing removeStubsFromRunpathHook.bash (hostOffset=-1) (targetOffset=0)
nix-shell> source: added removeStubsFromRunpathHookRegistration to prePhases
nix-shell> Sourcing fix-elf-files.sh
nix-shell> Running phase: buildPhase