diff --git a/infra/k8s/argo/gpu-test-pipeline-template.yaml b/infra/k8s/argo/gpu-test-pipeline-template.yaml index ac3497bee..caff57814 100644 --- a/infra/k8s/argo/gpu-test-pipeline-template.yaml +++ b/infra/k8s/argo/gpu-test-pipeline-template.yaml @@ -61,6 +61,8 @@ spec: dependencies: [gpu-warmup] # ── gpu-warmup: trigger H100 autoscale ── + # Requests GPU to force autoscaler to provision node, then releases it. + # Uses the runtime image (not busybox) for Argo emissary executor compat. - name: gpu-warmup nodeSelector: k8s.scaleway.com/pool-name: "{{workflow.parameters.gpu-pool}}" @@ -72,12 +74,13 @@ spec: operator: Exists effect: NoSchedule container: - image: busybox:1.37 + image: gitlab-registry.foxhunt.svc.cluster.local:5000/root/foxhunt/foxhunt-runtime:latest command: ["/bin/sh", "-c"] args: - | echo "GPU warmup: triggering node autoscale..." - echo "GPU node scheduled, exiting to free resources" + nvidia-smi --query-gpu=name,memory.total --format=csv,noheader || true + echo "GPU node ready, releasing for compile-and-test" resources: requests: nvidia.com/gpu: "1"