feat(tasks): add cluster:gpu task and wire e2e:gpu to use it

Author: elezar

e2e/python/conftest.py

@@ -96,10 +96,10 @@ def _run(sandbox: Sandbox, code: str) -> tuple[int, str, str]:
 
 @pytest.fixture(scope="session")
 def gpu_sandbox_spec() -> datamodel_pb2.SandboxSpec:
-    image = os.environ.get(
-        "OPENSHELL_E2E_GPU_IMAGE",
-        "ghcr.io/nvidia/openshell-community/sandboxes/nvidia-gpu:latest",
-    )
+    # Empty string defers image resolution to the server, which substitutes
+    # the configured default sandbox image.  Set OPENSHELL_E2E_GPU_IMAGE to
+    # override (e.g. a locally-built or registry-mirrored image).
+    image = os.environ.get("OPENSHELL_E2E_GPU_IMAGE", "")
     return datamodel_pb2.SandboxSpec(
         gpu=True,
         template=datamodel_pb2.SandboxTemplate(image=image),

tasks/cluster.toml

@@ -7,6 +7,11 @@
 description = "Bootstrap or incremental deploy (creates cluster if needed, rebuilds changed components)"
 run = "tasks/scripts/cluster.sh"
 
+["cluster:gpu"]
+description = "Bootstrap or incremental deploy with NVIDIA GPU passthrough enabled"
+env = { CLUSTER_GPU = "1" }
+run = "tasks/scripts/cluster.sh"
+
 ["cluster:build:full"]
 description = "Build and deploy local k3s cluster with OpenShell"
 depends = [

tasks/scripts/cluster-bootstrap.sh

@@ -243,6 +243,10 @@ fi
 
 DEPLOY_CMD=(openshell gateway start --name "${CLUSTER_NAME}" --port "${GATEWAY_PORT}")
 
+if [ "${CLUSTER_GPU:-0}" = "1" ]; then
+  DEPLOY_CMD+=(--gpu)
+fi
+
 if [ -n "${GATEWAY_HOST:-}" ]; then
   DEPLOY_CMD+=(--gateway-host "${GATEWAY_HOST}")
 

tasks/test.toml

@@ -43,6 +43,6 @@ run = "uv run pytest -o python_files='test_*.py' -m 'not gpu' -n ${E2E_PARALLEL:
 
 ["e2e:python:gpu"]
 description = "Run Python GPU e2e tests"
-depends = ["python:proto", "cluster"]
+depends = ["python:proto", "cluster:gpu"]
 env = { UV_NO_SYNC = "1", PYTHONPATH = "python" }
 run = "uv run pytest -o python_files='test_*.py' -m gpu -n ${E2E_PARALLEL:-1} e2e/python"