Skip to content

Commit d781534

Browse files
committed
Add monitor skill for persistent job tracking
Signed-off-by: Kai Xu <kaix@nvidia.com>
1 parent 202c3d3 commit d781534

File tree

4 files changed

+120
-60
lines changed

4 files changed

+120
-60
lines changed

.claude/skills/deployment/SKILL.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ If a cluster config exists (`~/.config/modelopt/clusters.yaml` or `.claude/clust
193193

194194
3. **Deploy based on remote environment:**
195195

196-
- **SLURM** — see `skills/common/slurm-setup.md` for job script templates (container setup, account/partition discovery). The server command inside the container is the same as Step 4 (e.g., `python -m vllm.entrypoints.openai.api_server --model <path> --quantization modelopt`). Use `remote_submit_job` and `remote_poll_job` to manage the job. Get the node hostname from `squeue -j $JOBID -o %N`.
196+
- **SLURM** — see `skills/common/slurm-setup.md` for job script templates (container setup, account/partition discovery). The server command inside the container is the same as Step 4 (e.g., `python -m vllm.entrypoints.openai.api_server --model <path> --quantization modelopt`). After submitting, register the job and set up monitoring per the **monitor skill**. Get the node hostname from `squeue -j $JOBID -o %N`.
197197

198198
- **Bare metal / Docker** — use `remote_run` to start the server directly:
199199

.claude/skills/evaluation/SKILL.md

Lines changed: 16 additions & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -225,64 +225,24 @@ After the dry-run, check the output from `nel` for any problems with the config.
225225

226226
**Monitoring Progress**
227227

228-
After job submission, you can monitor progress using:
228+
After job submission, register the job and set up monitoring per the **monitor skill**.
229229

230-
1. **Check job status:**
230+
**NEL-specific diagnostics** (for debugging failures):
231231

232-
```bash
233-
nel status <invocation_id>
234-
nel info <invocation_id>
235-
```
236-
237-
2. **Stream logs** (Local execution only):
238-
239-
```bash
240-
nel logs <invocation_id>
241-
```
242-
243-
Note: `nel logs` is not supported for SLURM execution.
244-
245-
3. **Inspect logs via SSH** (SLURM workaround):
246-
247-
When `nel logs` is unavailable (SLURM), use SSH to inspect logs directly:
248-
249-
First, get log locations:
250-
251-
```bash
252-
nel info <invocation_id> --logs
253-
```
254-
255-
Then, use SSH to view logs:
256-
257-
**Check server deployment logs:**
258-
259-
```bash
260-
ssh <username>@<hostname> "tail -100 <log path from `nel info <invocation_id> --logs`>/server-<slurm_job_id>-*.log"
261-
```
262-
263-
Shows vLLM server startup, model loading, and deployment errors (e.g., missing wget/curl).
264-
265-
**Check evaluation client logs:**
266-
267-
```bash
268-
ssh <username>@<hostname> "tail -100 <log path from `nel info <invocation_id> --logs`>/client-<slurm_job_id>.log"
269-
```
270-
271-
Shows evaluation progress, task execution, and results.
272-
273-
**Check SLURM scheduler logs:**
274-
275-
```bash
276-
ssh <username>@<hostname> "tail -100 <log path from `nel info <invocation_id> --logs`>/slurm-<slurm_job_id>.log"
277-
```
278-
279-
Shows job scheduling, health checks, and overall execution flow.
280-
281-
**Search for errors:**
282-
283-
```bash
284-
ssh <username>@<hostname> "grep -i 'error\|warning\|failed' <log path from `nel info <invocation_id> --logs`>/*.log"
285-
```
232+
```bash
233+
# Quick status check
234+
nel status <invocation_id>
235+
nel info <invocation_id>
236+
237+
# Get log paths
238+
nel info <invocation_id> --logs
239+
240+
# Inspect logs via SSH
241+
ssh <user>@<host> "tail -100 <log_path>/server-<slurm_job_id>-*.log" # deployment errors
242+
ssh <user>@<host> "tail -100 <log_path>/client-<slurm_job_id>.log" # evaluation errors
243+
ssh <user>@<host> "tail -100 <log_path>/slurm-<slurm_job_id>.log" # scheduling/walltime
244+
ssh <user>@<host> "grep -i 'error\|failed' <log_path>/*.log" # search all logs
245+
```
286246

287247
---
288248

.claude/skills/monitor/SKILL.md

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,102 @@
1+
---
2+
name: monitor
3+
description: Monitor submitted jobs (PTQ, evaluation, deployment) on SLURM clusters. Use when the user asks "check job status", "is my job done", "monitor my evaluation", "what's the status of the PTQ", "check on job 12345", or after any skill submits a long-running job. Also triggers on "nel status", "squeue", or any request to check progress of a previously submitted job.
4+
---
5+
6+
# Job Monitor
7+
8+
Monitor jobs submitted to SLURM clusters — PTQ quantization, NEL evaluation, model deployment, or raw SLURM jobs.
9+
10+
## When to use
11+
12+
1. **Auto-monitor** — another skill (PTQ, evaluation, deployment) just submitted a job. Register the job and set up monitoring immediately.
13+
2. **User-initiated** — user asks about a job status, possibly in a new conversation. Check the registry, identify the job, and report.
14+
15+
---
16+
17+
## Job Registry
18+
19+
All active jobs are tracked in `.claude/active_jobs.json`. This file is the single source of truth for what's being monitored.
20+
21+
```json
22+
[
23+
{
24+
"type": "nel",
25+
"id": "<invocation_id or slurm_job_id>",
26+
"host": "<cluster_hostname>",
27+
"user": "<ssh_user>",
28+
"submitted": "YYYY-MM-DD HH:MM",
29+
"description": "<what this job does>",
30+
"last_status": "<last known status>"
31+
}
32+
]
33+
```
34+
35+
`type` is one of: `nel`, `slurm`, `launcher`.
36+
37+
---
38+
39+
## On Job Submission
40+
41+
Every time a job is submitted (by any skill or manually):
42+
43+
1. **Add an entry** to `.claude/active_jobs.json`. Create the file if it doesn't exist.
44+
2. **Set up a durable recurring cron** (if one isn't already running) that polls all registered jobs every 15 minutes. The cron prompt should: read the registry, check each job, report state changes to the user, remove completed jobs, and delete itself when the registry is empty.
45+
46+
Always do both steps. Don't try to predict job duration.
47+
48+
---
49+
50+
## On Cron Fire / Status Check
51+
52+
Whether triggered by the cron or by the user asking "check status":
53+
54+
1. **Read the registry** from `.claude/active_jobs.json`
55+
2. **Check each job** using the appropriate method (see below)
56+
3. **Report only state changes** — compare against `last_status` in registry
57+
4. **Update `last_status`** in the registry
58+
5. **Remove completed jobs** — any job in a terminal state (COMPLETED, FAILED, CANCELLED, KILLED)
59+
6. **If registry is empty** — delete the recurring cron
60+
61+
---
62+
63+
## How to Check Each Job Type
64+
65+
### NEL jobs (`type: nel`)
66+
67+
- **Check:** `nel status <id>`
68+
- **On completion:** `nel info <id>` to fetch results
69+
- **On failure:** `nel info <id> --logs` then inspect server/client/SLURM logs via SSH
70+
71+
### Launcher jobs (`type: launcher`)
72+
73+
- **Check:** Tail the launcher's background output file for key events
74+
- **Key events:** experiment ID, SLURM job ID, container import, calibration progress, export path, final status
75+
- **On failure:** Look for `Traceback`, `Error`, or `FAILED` in the output
76+
77+
### Raw SLURM jobs (`type: slurm`)
78+
79+
- **Check:** `ssh <host> "squeue -j <id> -h -o '%T %M %R'"` — if empty, job left the queue
80+
- **On completion:** `ssh <host> "sacct -j <id> --format=State,ExitCode,Elapsed -n"`
81+
- **On failure:** Check the job's output log file
82+
83+
---
84+
85+
## Identifying Jobs (user-initiated, no ID given)
86+
87+
When the user asks about a job without specifying an ID, check in order:
88+
89+
1. `.claude/active_jobs.json` — most reliable, has context
90+
2. `nel ls runs --since 1d` — recent NEL runs
91+
3. `ssh <host> "squeue -u <user>"` — active SLURM jobs
92+
4. `ls -lt tools/launcher/experiments/cicd/ | head -10` — recent launcher experiments
93+
94+
---
95+
96+
## Reporting Guidelines
97+
98+
- **Report state changes proactively** — PENDING → RUNNING, or job completes
99+
- **Aggregate multiple jobs** — "2 of 4 completed (MMLU-Pro: 42.3%, GSM8K: 67.1%), 1 running, 1 pending"
100+
- **Summarize, don't echo** — interpret events ("Calibration complete, exporting checkpoint") not raw logs
101+
- **On failure, diagnose immediately** — check logs and report root cause without waiting for user to ask
102+
- **Minimize noise** — don't report "still running" unless the user is actively asking

.claude/skills/ptq/SKILL.md

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,7 @@ For SLURM, see `skills/common/slurm-setup.md` and `references/slurm-setup-ptq.md
100100

101101
### Monitoring
102102

103-
- **Launcher**: blocks and tails logs automatically
104-
- **SLURM (manual)**: poll with `squeue -u $USER` + `sleep` (not cron or background tasks)
105-
- **Local**: watch stdout
103+
After job submission, register the job and set up monitoring per the **monitor skill**.
106104

107105
## Step 5 — Verify output
108106

0 commit comments

Comments
 (0)