Skip to content

Commit e2f7cd6

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

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
@@ -195,7 +195,7 @@ If a cluster config exists (`~/.config/modelopt/clusters.yaml` or `.claude/clust
195195

196196
3. **Deploy based on remote environment:**
197197

198-
- **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`.
198+
- **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`.
199199

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

.claude/skills/evaluation/SKILL.md

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

258258
**Monitoring Progress**
259259

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

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

264-
```bash
265-
nel status <invocation_id>
266-
nel info <invocation_id>
267-
```
268-
269-
2. **Stream logs** (Local execution only):
270-
271-
```bash
272-
nel logs <invocation_id>
273-
```
274-
275-
Note: `nel logs` is not supported for SLURM execution.
276-
277-
3. **Inspect logs via SSH** (SLURM workaround):
278-
279-
When `nel logs` is unavailable (SLURM), use SSH to inspect logs directly:
280-
281-
First, get log locations:
282-
283-
```bash
284-
nel info <invocation_id> --logs
285-
```
286-
287-
Then, use SSH to view logs:
288-
289-
**Check server deployment logs:**
290-
291-
```bash
292-
ssh <username>@<hostname> "tail -100 <log path from `nel info <invocation_id> --logs`>/server-<slurm_job_id>-*.log"
293-
```
294-
295-
Shows vLLM server startup, model loading, and deployment errors (e.g., missing wget/curl).
296-
297-
**Check evaluation client logs:**
298-
299-
```bash
300-
ssh <username>@<hostname> "tail -100 <log path from `nel info <invocation_id> --logs`>/client-<slurm_job_id>.log"
301-
```
302-
303-
Shows evaluation progress, task execution, and results.
304-
305-
**Check SLURM scheduler logs:**
306-
307-
```bash
308-
ssh <username>@<hostname> "tail -100 <log path from `nel info <invocation_id> --logs`>/slurm-<slurm_job_id>.log"
309-
```
310-
311-
Shows job scheduling, health checks, and overall execution flow.
312-
313-
**Search for errors:**
314-
315-
```bash
316-
ssh <username>@<hostname> "grep -i 'error\|warning\|failed' <log path from `nel info <invocation_id> --logs`>/*.log"
317-
```
264+
```bash
265+
# Quick status check
266+
nel status <invocation_id>
267+
nel info <invocation_id>
268+
269+
# Get log paths
270+
nel info <invocation_id> --logs
271+
272+
# Inspect logs via SSH
273+
ssh <user>@<host> "tail -100 <log_path>/server-<slurm_job_id>-*.log" # deployment errors
274+
ssh <user>@<host> "tail -100 <log_path>/client-<slurm_job_id>.log" # evaluation errors
275+
ssh <user>@<host> "tail -100 <log_path>/slurm-<slurm_job_id>.log" # scheduling/walltime
276+
ssh <user>@<host> "grep -i 'error\|failed' <log_path>/*.log" # search all logs
277+
```
318278

319279
---
320280

.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
@@ -118,9 +118,7 @@ For SLURM, see `skills/common/slurm-setup.md` and `references/slurm-setup-ptq.md
118118

119119
### Monitoring
120120

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

125123
## Step 5 — Verify output
126124

0 commit comments

Comments
 (0)