agent-gpu-skills

面向 GPU kernel 开发的 Agent Skill 集合，提供 CUDA/PTX 官方文档快照，以及 Triton、Gluon、CUTLASS、CuTeDSL 和 TileLang 的本地源码检索入口。

Skill	边界	适用场景
cuda-skill	CUDA、PTX 和 NVIDIA 工具	查询 ISA、CUDA API、架构特性、Nsight 与 Compute Sanitizer
cutlass-skill	CUTLASS、CuTe 和 CuTeDSL	编写和调试 CUTLASS/CuTe kernel，检索 C++ 与 Python DSL 示例
tilelang-skill	TileLang DSL、JIT、compiler 与多后端	编写和调试 TileLang kernel，追踪 lowering、生成代码、autotune 与 profiler
triton-skill	Triton 和 Gluon	编写和调试 Python GPU kernel，检索教程、生产内核与编译器源码

快速开始

git clone https://github.com/slowlyC/agent-gpu-skills.git
cd agent-gpu-skills
bash update-repos.sh
bash install.sh --agent codex

update-repos.sh 默认创建 shallow sparse checkout。安装器复制 SKILL.md，并将大型参考资料和上游源码软链接到安装目录。

已有 Triton、CUTLASS 或 TileLang checkout 时，可以直接复用：

TRITON_REPO=/path/to/triton \
CUTLASS_REPO=/path/to/cutlass \
TILELANG_REPO=/path/to/tilelang \
bash install.sh --agent codex

安装

安装目标

工具	Skill 目录	命令	验证状态
Cursor	~/.cursor/skills/	bash install.sh	已验证
Claude Code	~/.claude/skills/	bash install.sh --agent claude	未完整验证
Codex	~/.codex/skills/	bash install.sh --agent codex	安装器已验证
Gemini CLI	~/.gemini/skills/	bash install.sh --agent gemini	未完整验证
Qoder	~/.qoder/skills/	bash install.sh --agent qoder	未完整验证

Codex 的配置文件是 ~/.codex/config.toml，Skill 安装目录是 ~/.codex/skills/。设置了 CODEX_HOME 时，安装器使用 $CODEX_HOME/skills/。

安装模式

默认使用混合模式：

• Skill 目录和 SKILL.md 是普通目录与文件，便于各工具发现 Skill。
• CUDA 参考资料、quick reference 和上游源码使用软链接，避免重复占用空间。
• Triton、CUTLASS 和 TileLang checkout 保存在 third_party/，不作为 Git submodule 提交。

无法使用软链接时，可以完整复制：

bash install.sh --agent codex --copy

全量复制也会复制 Triton、CUTLASS 和 TileLang checkout。指向完整源码仓库时，先确认目标磁盘空间。

安装器不会删除普通文件或目录，也不会覆盖指向其他位置的软链接。旧安装中的链接失效时，先检查迁移内容，再显式使用修复模式：

bash install.sh --agent codex --repair-links

--repair-links 只替换 Skill 安装目录中的软链接，不删除普通文件或目录。

安装单个 Skill

bash install.sh --agent codex --skill cuda-skill
bash install.sh --agent codex --skill cutlass-skill
bash install.sh --agent codex --skill tilelang-skill
bash install.sh --agent codex --skill triton-skill
bash install.sh --target-dir /custom/skills --skill cuda-skill

更新

更新默认的 sparse checkout：

bash update-repos.sh
bash update-repos.sh triton
bash update-repos.sh cutlass
bash update-repos.sh tilelang

只有编译完整上游、运行完整测试集或检查 sparse 路径之外的代码时，才展开完整工作树：

bash update-repos.sh all --full

--full 只展开三个上游仓库的主工作树，不主动管理嵌套 submodule。默认 TileLang sparse checkout 面向源码检索，不包含 3rdparty/ 中的 TVM、CUTLASS、Composable Kernel 等依赖。需要编译 TileLang 时，优先通过 TILELANG_REPO 指向完整开发 checkout，并遵循上游 .agents/skills/tilelang-build/SKILL.md 中与当前版本匹配的构建流程。

third_party/UPSTREAMS.toml 记录最后一次通过本仓库验证的上游 commit。update-repos.sh 负责拉取源码，但不会自动改写该记录；检查新版本后再更新 commit，避免把未经验证的上游状态当作已接受版本。

CUDA 文档通过带有 PEP 723 依赖声明的抓取脚本维护，需要本机安装 uv：

uv run scripts/scrape_docs.py all \
  --output-dir /tmp/cuda-docs-staging \
  --force

diff -qr skills/cuda-skill/references/ptx-docs \
  /tmp/cuda-docs-staging/ptx-docs

抓取脚本不会删除已有内容。先写入暂存目录，核对版本、文件数量和代表性页面后，再合并到正式快照。

验证

python3 scripts/validate_repo.py --require-sources
python3 scripts/validate_cuda_skill.py

validate_repo.py 检查四个 Skill、维护脚本、上游 commit、代表性源码路径和已知的目录漂移。需要检查安装结果时，传入实际 Skill 根目录：

python3 scripts/validate_repo.py \
  --require-sources \
  --install-root ~/.codex/skills

目录结构

agent-gpu-skills/
├── README.md
├── LICENSE
├── install.sh
├── update-repos.sh
├── scripts/
│   ├── scrape_docs.py
│   ├── validate_cuda_skill.py
│   └── validate_repo.py
├── skills/
│   ├── cuda-skill/
│   │   ├── SKILL.md
│   │   └── references/
│   ├── cutlass-skill/
│   │   ├── SKILL.md
│   │   └── quick-reference.md
│   ├── tilelang-skill/
│   │   ├── SKILL.md
│   │   └── quick-reference.md
│   └── triton-skill/
│       ├── SKILL.md
│       └── quick-reference.md
└── third_party/
    ├── README.md
    ├── UPSTREAMS.toml
    ├── cutlass/
    ├── tilelang/
    └── triton/

skills/ 保存本仓库维护的 Skill 和参考资料。third_party/ 保存可重建的上游 checkout，并通过 .gitignore 排除源码内容。

CUDA 文档快照

文档	文件数	版本
PTX ISA	489	9.3
CUDA Runtime API	106	13.3.1
CUDA Driver API	135	13.3.1
CUDA Programming Guide	44	13.3
CUDA C++ Best Practices Guide	73	CUDA 13.3 文档集
Nsight Compute	9	2026.2.1
Nsight Systems	5	2026.3

版本、来源和文件清单见 skills/cuda-skill/references/MANIFEST.md。

上游源码范围

cutlass-skill 默认检索以下内容：

• python/CuTeDSL/ 中的 CuTeDSL 实现
• examples/python/CuTeDSL/ 中的全部 Python kernel、教程与集成示例
• examples/ 中的 CUTLASS C++ 示例
• include/cute/ 与 include/cutlass/ 中的头文件

triton-skill 默认检索以下内容：

• python/tutorials/ 下的全部 Triton 教程
• python/tutorials/gluon/ 下的全部 Gluon 教程
• python/examples/ 下的全部完整示例
• triton_kernels 中的 matmul、reduction、top-k、SwiGLU 与数值工具
• Triton language、runtime、compiler、IR 和 lowering 源码

tilelang-skill 默认检索以下内容：

• docs/ 下的编程指南、工具说明和 compiler/runtime 文档
• examples/ 下的全部 kernel 与端到端 workload
• tilelang/ 下的 language、JIT、autotune、profiler、layout 与多后端 Python 源码
• src/ 下的 compiler transform、target codegen 与 runtime 源码
• testing/python/ 和上游 build Skill，用于查找当前验证与构建合同

详细路径分别见三个源码型 Skill 的 quick-reference.md。

许可

本仓库原创脚本和文档使用 MIT License。skills/cuda-skill/references/ 中的 CUDA 文档内容归 NVIDIA Corporation 所有，third_party/ 中的 Triton、CUTLASS 和 TileLang 源码遵循各自的上游许可。