diff --git a/.github/workflows/doc_build.yml b/.github/workflows/doc_build.yml index 2903e9cf8..125fa83e5 100644 --- a/.github/workflows/doc_build.yml +++ b/.github/workflows/doc_build.yml @@ -41,6 +41,7 @@ jobs: run: | python -m pip install torch python -m pip install -e . + pip install -r dev-requirements.txt cd docs python -m pip install -r requirements.txt - name: Build docs diff --git a/docs/source/api_ref_sparsity.rst b/docs/source/api_ref_sparsity.rst index 9417d9bef..8023d0bac 100644 --- a/docs/source/api_ref_sparsity.rst +++ b/docs/source/api_ref_sparsity.rst @@ -3,7 +3,7 @@ ================ torchao.sparsity ================ - +.. automodule:: torchao.sparsity .. currentmodule:: torchao.sparsity .. autosummary:: @@ -12,3 +12,7 @@ torchao.sparsity WandaSparsifier PerChannelNormObserver + apply_sparse_semi_structured + apply_fake_sparsity + + diff --git a/docs/source/index.rst b/docs/source/index.rst index fb1649fa4..cea80772d 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -92,9 +92,9 @@ with more content coming soon. :maxdepth: 1 :caption: API Reference - api_ref_intro api_ref_sparsity + api_ref_intro api_ref_quantization api_ref_dtypes - .. +.. api_ref_kernel