Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions .claude/agents/ad-run-agent.md
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
---
name: ad-run-agent
description: Run AutoDeploy build and run command for a given model using gpu-shell
tools: Read, Grep, Glob, Bash, Write, Edit, gpu-shell
description: Run AutoDeploy build and run command for a given model
tools: Read, Grep, Glob, Bash, Write, Edit
model: sonnet
---

Run a model with AutoDeploy using gpu-shell. If you are not given the model-id and a short description of the run, ask the user/caller first.
Run a model with AutoDeploy. If you are not given the model-id and a short description of the run, ask the user/caller first.

## Prerequisites — Model Registry Entry

Expand Down Expand Up @@ -43,7 +43,7 @@ Look up the model in `examples/auto_deploy/model_registry/models.yaml` and find

**Step 2 — Check GPU availability via `nvidia-smi`:**

Run via gpu-shell:
Run via Bash:
```bash
nvidia-smi --query-gpu=index,memory.used,utilization.gpu --format=csv,noheader,nounits
```
Expand All @@ -57,7 +57,7 @@ A GPU is considered **free** if its memory usage is below ~1000 MiB and utilizat

### 1. Run the AD build and run command

Execute via gpu-shell since it requires GPU access:
Execute via Bash:
```bash
CUDA_VISIBLE_DEVICES=<SELECTED_GPUS> AD_DUMP_GRAPHS_DIR=<AD_DUMP_GRAPHS_DIR> \
python examples/auto_deploy/build_and_run_ad.py \
Expand Down Expand Up @@ -120,4 +120,4 @@ Then append the first entry.
- Always ask for the run description from the caller before starting.
- Before running, verify the model has a valid entry in `examples/auto_deploy/model_registry/models.yaml`. If missing, create one.
- If it failed, show the relevant error and suggest next steps.
- Remember to use your own tools — Read, Grep, Glob, Bash, Write, Edit, gpu-shell
- Remember to use your own tools — Read, Grep, Glob, Bash, Write, Edit
2 changes: 1 addition & 1 deletion .codex/agents/ad_debug_agent.toml
Original file line number Diff line number Diff line change
Expand Up @@ -39,5 +39,5 @@ Common pitfalls:
- HF safetensors weights may not match what the AD custom modeling code expects. Verify load hooks and inspect the safetensors index json when relevant.
- Custom model module hierarchies may differ from the checkpoint hierarchy. In that case, update the AD custom modeling code to match the expected hierarchy.

Use your own tools to read files, search, edit, and run shell commands. If GPU access is required, use the available execution path in this environment rather than assuming a separate gpu-shell tool exists.
Use your own tools to read files, search, edit, and run shell commands.
"""
2 changes: 1 addition & 1 deletion .codex/agents/ad_run_agent.toml
Original file line number Diff line number Diff line change
Expand Up @@ -64,5 +64,5 @@ If the worklog does not exist, create it with this header first:

Notes:
- If the run fails, show the relevant error and suggest next steps.
- Use the available shell execution path in this environment for GPU-capable commands rather than assuming a separate gpu-shell tool exists.
- Use the standard shell execution path for GPU-capable commands.
"""
Loading