cuda-webdoc-search
Compare original and translation side by side
🇺🇸
Original
English🇨🇳
Translation
ChineseCUDA-X Documentation Search Guide
CUDA-X文档搜索指南
Overview
概述
Search and discover APIs across CUDA-X library documentation by querying Sphinx inventory files. Use this skill when you need to find specific CUDA library functions, check available APIs, or get documentation links.
通过查询Sphinx清单文件,搜索并发现CUDA-X库文档中的各类API。当你需要查找特定CUDA库函数、查看可用API或获取文档链接时,可使用该工具。
Setup
安装配置
bash
uv tool install "cuda-webdoc-search @ git+https://github.com/ultimatile/cuda-x-skills.git#subdirectory=skills/cuda-webdoc-search"After installation, is available as a standalone command. For one-shot usage without installing:
cwsbash
uvx --from "cuda-webdoc-search @ git+https://github.com/ultimatile/cuda-x-skills.git#subdirectory=skills/cuda-webdoc-search" cws search cublas --statsbash
uv tool install "cuda-webdoc-search @ git+https://github.com/ultimatile/cuda-x-skills.git#subdirectory=skills/cuda-webdoc-search"安装完成后,将作为独立命令可用。若无需安装即可一次性使用:
cwsbash
uvx --from "cuda-webdoc-search @ git+https://github.com/ultimatile/cuda-x-skills.git#subdirectory=skills/cuda-webdoc-search" cws search cublas --statsWhen to Use
使用场景
- Finding CUDA library APIs (e.g., "what SVD functions does cuTensorNet have?")
- Checking if a specific function exists in a library
- Getting documentation URLs for CUDA APIs
- Exploring available operations in a library
- 查找CUDA库API(例如:“cuTensorNet有哪些SVD函数?”)
- 检查某个特定函数是否存在于某库中
- 获取CUDA API的文档URL
- 探索库中的可用操作
Quick Reference
快速参考
Check Available Domains First
先查看可用领域
Before searching, check what domains (API types) are available:
bash
cws search <library> --statsThis returns domain counts: (C++ APIs), (C APIs), (Python bindings), (doc labels).
cppcpystd搜索前,先查看可用的领域(API类型):
bash
cws search <library> --stats该命令会返回领域数量:(C++ API)、(C API)、(Python绑定)、(文档标签)。
cppcpystdSearch for APIs
搜索API
bash
cws search <library> --domains <domain> --keywords "<terms>" --fuzzy --limit 20bash
cws search <library> --domains <domain> --keywords "<terms>" --fuzzy --limit 20Keyword Syntax (fzf subset)
关键字语法(fzf子集)
- Space-separated terms are AND (all must match):
--keywords "SVD batch" - Use for OR (requires shell quoting):
|--keywords "SVD | QR" - AND binds tighter than OR: = (a AND b) OR c
--keywords "a b | c" - Note: Only AND/OR via is supported. fzf operators
|,^,',!are not available.$
- 空格分隔的术语表示AND(必须全部匹配):
--keywords "SVD batch" - 使用表示OR(需要shell引号包裹):
|--keywords "SVD | QR" - AND的优先级高于OR:= (a AND b) OR c
--keywords "a b | c" - 注意:仅支持通过实现AND/OR逻辑。fzf的
|、^、'、!运算符暂不支持。$
Available Libraries
可用库
See for the full list. Common ones:
registry.toml- - cuTensorNet, cuStateVec (quantum/tensor network)
cuquantum - - BLAS operations
cublas - - Dense/sparse solvers
cusolver - - Sparse matrix operations
cusparse - - FFT operations
cufft - - Deep learning primitives
cudnn - - GEMM templates
cutlass - - Math functions
cuda_math
完整列表请查看。常用库包括:
registry.toml- - cuTensorNet、cuStateVec(量子/张量网络)
cuquantum - - BLAS运算
cublas - - 稠密/稀疏求解器
cusolver - - 稀疏矩阵运算
cusparse - - FFT运算
cufft - - 深度学习原语
cudnn - - GEMM模板
cutlass - - 数学函数
cuda_math
Workflow
使用流程
- Identify the library: Determine which CUDA library likely contains the API
- Check domains: Run to see available domain types
--stats - Search: Use with
--keywordsfor flexible matching--fuzzy - Filter by domain: Use for C++ APIs,
--domains cppfor C APIs--domains c - Extract details: Use to get full documentation content
cws get
- 确定库:判断哪个CUDA库可能包含所需API
- 查看领域:运行查看可用领域类型
--stats - 搜索:结合和
--keywords进行灵活匹配--fuzzy - 按领域过滤:使用筛选C++ API,
--domains cpp筛选C API--domains c - 提取详情:使用获取完整文档内容
cws get
Examples
使用示例
Find tensor decomposition APIs in cuTensorNet
在cuTensorNet中查找张量分解API
bash
cws search cuquantum --statsbash
cws search cuquantum --statsShows: cpp:3179, py:1172, std:3378, c:4
输出:cpp:3179, py:1172, std:3378, c:4
cws search cuquantum --domains cpp --keywords "SVD" --fuzzy --limit 10
cws search cuquantum --domains cpp --keywords "SVD" --fuzzy --limit 10
Returns: cutensornetTensorSVD, etc. (functions ranked above enumerators)
返回:cutensornetTensorSVD等(函数排序优先于枚举器)
cws search cuquantum --domains cpp --keywords "SVD | QR" --fuzzy
cws search cuquantum --domains cpp --keywords "SVD | QR" --fuzzy
Returns: entries matching SVD OR QR
返回:匹配SVD或QR的条目
undefinedundefinedSearch across multiple libraries
跨多个库搜索
bash
undefinedbash
undefined"Where is SVD in CUDA-X?" — search cuSOLVER and cuDSS together
“SVD在CUDA-X的哪个库中?”——同时搜索cuSOLVER和cuDSS
cws search cusolver cudss --keywords "svd" --fuzzy --limit 10
cws search cusolver cudss --keywords "svd" --fuzzy --limit 10
Mix any number of sources
混合任意数量的源库
cws search cusolver cusparse cudss --keywords "solve" --fuzzy
Multi-source output uses `"sources"` (list) instead of `"source"` (string), and `"total_found"` becomes a per-source dict.cws search cusolver cusparse cudss --keywords "solve" --fuzzy
多源输出会使用`sources`(列表)替代`source`(字符串),`total_found`会变为按源库统计的字典。Find GEMM functions in cuBLAS
在cuBLAS中查找GEMM函数
bash
cws search cublas --statsbash
cws search cublas --statsShows: std:36 (only doc labels, no cpp/c domain)
输出:std:36(仅包含文档标签,无cpp/c领域)
cws search cublas --domains std --keywords "gemm"
cws search cublas --domains std --keywords "gemm"
Returns: cublas-t-gemm, cublas-t-gemmex, etc. (doc section labels)
返回:cublas-t-gemm、cublas-t-gemmex等(文档章节标签)
undefinedundefinedExtract API documentation details
提取API文档详情
After finding a function URL, extract its full documentation:
bash
cws get <url> --section <function_name>Example:
bash
cws get \
"https://docs.nvidia.com/cuda/cuquantum/latest/cutensornet/api/functions.html" \
--section "cutensornetTensorSVD"Output is a brace-delimited text tree:
{
cutensornetTensorSVD
{
cutensornetStatus_t cutensornetTensorSVD {
const cutensornetHandle_t handle
const cutensornetTensorDescriptor_t descTensorIn
...
}
Performs SVD decomposition of a tensor. { ... }
Parameters { ... }
}
}找到函数URL后,提取其完整文档:
bash
cws get <url> --section <function_name>示例:
bash
cws get \
"https://docs.nvidia.com/cuda/cuquantum/latest/cutensornet/api/functions.html" \
--section "cutensornetTensorSVD"输出为大括号分隔的文本树:
{
cutensornetTensorSVD
{
cutensornetStatus_t cutensornetTensorSVD {
const cutensornetHandle_t handle
const cutensornetTensorDescriptor_t descTensorIn
...
}
Performs SVD decomposition of a tensor. { ... }
Parameters { ... }
}
}List available sources
列出可用源库
bash
cws list # table format
cws list --json # JSON formatbash
cws list # 表格格式
cws list --json # JSON格式Audit registry health
检查注册表健康状态
bash
cws audit # audit all sources
cws audit --source cublas # audit single sourcebash
cws audit # 检查所有源库
cws audit --source cublas # 检查单个源库Output Format
输出格式
cws search
cws search
JSON output includes:
Single source:
- : Library name (string)
source - : Total APIs in filtered domains
total_found - : APIs matching keywords
filtered_count - : List of matching APIs with
candidates(name),group,url,domainrole
Multi-source ():
cws search A B- : Library names (list)
sources - : Per-source totals (dict)
total_found - : Total APIs matching keywords across all sources
filtered_count - : Merged list (k-way merge by score if
candidates, else concatenated)--fuzzy
JSON输出包含:
单源库:
- : 库名称(字符串)
source - : 过滤后领域中的API总数
total_found - : 匹配关键字的API数量
filtered_count - : 匹配API列表,包含
candidates(名称)、group、url、domainrole
多源库():
cws search A B- : 库名称列表
sources - : 按源库统计的总数(字典)
total_found - : 所有源库中匹配关键字的API总数
filtered_count - : 合并后的列表(若使用
candidates则按得分排序,否则直接拼接)--fuzzy
cws get
cws get
Brace-delimited text tree where:
{denote hierarchy}- Text content is preserved
- Inline elements (code, links) are flattened to text
Options:
- : Extract only a specific section
--section <id> - : Extract only the main content area
--main-only
大括号分隔的文本树,其中:
{表示层级结构}- 文本内容保留原样
- 内联元素(代码、链接)转为纯文本
可选参数:
- : 仅提取特定章节
--section <id> - : 仅提取主内容区域
--main-only
Files
文件说明
- - Unified CLI entry point (
cli.py)cws - - Search APIs across library inventories
search.py - - Audit registry entries for endpoint health
audit.py - - Extract documentation content as text tree
get.py - - Data fetching for Sphinx/Doxygen sources
fetchers.py - - Search ranking and filtering logic
scoring.py - - Registry loading utility
registry.py - - Library metadata (URLs, doc types)
registry.toml
- - 统一CLI入口(
cli.py)cws - - 跨库清单搜索API
search.py - - 检查注册表条目的端点健康状态
audit.py - - 提取文档内容为文本树
get.py - - Sphinx/Doxygen源的数据获取
fetchers.py - - 搜索排序与过滤逻辑
scoring.py - - 注册表加载工具
registry.py - - 库元数据(URL、文档类型)
registry.toml