diff --git a/audit/internal/webui/api.go b/audit/internal/webui/api.go index 1c21425..dcf1ae0 100644 --- a/audit/internal/webui/api.go +++ b/audit/internal/webui/api.go @@ -267,9 +267,9 @@ func formatSplitTaskName(baseName, selectionLabel string) string { } func buildNvidiaTaskSet(target string, priority int, createdAt time.Time, params taskParams, baseName string, appRef *app.App, idPrefix string) ([]*Task, error) { - if !shouldSplitHomogeneousNvidiaTarget(target) || params.ParallelGPUs { - // Parallel mode (or non-splittable target): one task for all selected GPUs. - if params.ParallelGPUs && shouldSplitHomogeneousNvidiaTarget(target) { + if !shouldSplitHomogeneousNvidiaTarget(target) || params.ParallelGPUs || params.StaggerGPUStart { + // Parallel mode, ramp-up mode (or non-splittable target): one task for all selected GPUs. + if (params.ParallelGPUs || params.StaggerGPUStart) && shouldSplitHomogeneousNvidiaTarget(target) { // Resolve the selected GPU indices so ExcludeGPUIndices is applied. gpus, err := apiListNvidiaGPUs(appRef) if err != nil {