SuperAPI AI 工具集

cuda-webdoc-search

通用与其它

查看项目

更新于 10-16 01:48

技能介绍

用途:在 CUDA-X 库文档中检索 API。通过查询 Sphinx 清单文件,查找具体函数、确认 API 是否存在、获取文档链接。适用场景包括查找某库的 API(如“cuTensorNet 有哪些 SVD 函数”)、检查函数是否存在、探索库中的可用操作。

安装:uv tool install "cuda-webdoc-search @ git+https://github.com/ultimatile/cuda-x-skills.git#subdirectory=skills/cuda-webdoc-search",安装后可用独立命令 cws;也可用 uvx --from "..." cws search cublas --stats 一次性运行。

主要能力与用法:

  • 先查可用域:cws search <library> --stats,返回 cpp、c、py、std 等域计数。
  • 搜索 API:cws search <library> --domains <domain> --keywords "<terms>" --fuzzy --limit 20。关键词支持空格分隔的 AND,以及用 | 表示 OR(需 shell 引号),AND 优先级高于 OR;不支持 fzf 的 ^、'、!、$ 操作符。
  • 多库搜索:cws search cusolver cudss --keywords "svd" --fuzzy --limit 10,输出用 sources 列表,total_found 变为按源字典。
  • 提取文档详情:cws get <url> --section <function_name>,输出大括号分隔的文本树;支持 --section 与 --main-only。
  • 列出与审计:cws list(--json)、cws audit(--source cublas)。

可用库示例:cuquantum(cuTensorNet、cuStateVec)、cublas、cusolver、cusparse、cufft、cudnn、cutlass、cuda_math,完整列表见 registry.toml。

限制:仅支持 AND/OR 关键词语法;文档内容以文本树形式呈现,内联代码与链接被扁平化;未提供关于收费、平台或性能的信息。