快捷方式

文档

FBGEMM 和 FBGEMM_GPU 的源代码中都提供了丰富的注释,这些注释是这两个库最权威和最新的文档。

通用文档指南

在添加新的公共 API 方法时,应附带足够的文档。以下是为 FBGEMM 和 FBGEMM_GPU 代码编写文档的一些指南

  • 代码本身不是文档! 设身处地为需要理解你的代码的新开发者着想,让他们更容易上手。

  • 应为所有公共 API 方法添加文档。

  • 不要将文档作为单独的任务留到以后。相反,请与代码一起编写 docstrings。

  • 至少应包含:

    • 方法的描述。

    • 可传递给方法的参数和实参的描述。

    • 方法返回值的描述。

    • 使用示例、指向其他方法的链接以及方法调用限制。

特定文档指南

构建文档

注意: 最新文档构建说明已嵌入 FBGEMM 仓库中 setup_env.bash 下的一组脚本中。

构建 FBGEMM 和 FBGEMM_GPU 文档的一般步骤如下:

  1. 设置独立的构建环境。

  2. 构建 FBGEMM_GPU(CPU 变体)。

  3. 设置文档工具链。

  4. 运行文档构建脚本。

设置构建环境

按照 设置独立的构建环境 中的说明设置 Conda 环境。

构建 FBGEMM_GPU

为了正确构建文档,需要进行一次 FBGEMM_GPU 构建。请按照 安装构建工具 中的说明,然后按照 仅限 CPU 构建 的说明构建 FBGEMM_GPU(CPU 变体)。

设置文档工具链

# !! Run inside the Conda environment !!

# From the /fbgemm_gpu/ directory
cd docs

# Install Sphinx and other Python packages
pip install -r requirements.txt

# Install Doxygen and and other tools
conda install -c conda-forge --override-channels -y \
  doxygen \
  graphviz \
  make

构建文档

# Generate the C++ documentation, the Python documentation, and assemble
# together
make clean doxygen html

构建完成后,查看生成的文档

sphinx-serve -b build

检查文档

用于构建的相同命令也可用于检查,只需在命令前加上 SPHINX_LINT 标志即可

SPHINX_LINT=1 make clean doxygen html

出于技术原因,在启用检查的情况下运行 Sphinx 构建会导致文档组装不正确,这就是为什么检查与构建分开调用的原因。

偶尔,检查时可能会出现未解析的引用,其错误特征如下:

/opt/build/repo/fbgemm_gpu/docs/docstring of torch._ops.fbgemm.PyCapsule.jagged_2d_to_dense:1:py:class reference target not found: Tensor

如果这些错误是误报,则可以通过将它们添加到 nitpick.ignore 文件(与 Sphinx conf.py 位于同一目录下)来忽略它们

# Add in `{domain} {reference}` format, with space in between.
py:class Tensor

部署预览

当创建拉取请求时,FBGEMM 和 FBGEMM_GPU 文档的预览将由 Netlify 自动构建和部署。构建完成后,可以在以下位置找到部署预览:

https://deploy-preview-{PR NUMBER}--pytorch-fbgemm-docs.netlify.app/

文档

访问 PyTorch 全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并解答你的问题

查看资源