kfpytorch: make Elastic task_type dynamic so nnodes overrides flip single-pod vs distributed#9

Draft

codegen-sh[bot] wants to merge 3 commits into

codegen-bot/dynamic-elastic-task-type-override-a-1754596225

Commits on Aug 7, 2025

kfpytorch: make Elastic task_type dynamic so nnodes overrides flip single-pod (python-task) vs distributed (pytorch)\n\n- Override task_type property in PytorchElasticFunctionTask to derive from cu…
codegen-sh[bot]
committed
kfpytorch(elastic): force nnodes=1 in single-pod mode to avoid rendezvous misconfig from env overrides\n\n- When task_type resolves to python-task, ignore PET_NNODES and pass 1 to parse_min_max_nno…
codegen-sh[bot]
committed
kfpytorch(elastic): add detailed debug logs + loopback rdzv in single-pod mode\n\n- Log mode, nnodes_str (raw), parsed min/max, nproc_per_node, rdzv_backend, rdzv_endpoint, start_method\n- Derive s…
codegen-sh[bot]
committed