Install dev requirements

pytorch · May 7, 2024 · 49f1c1b · 49f1c1b
1 parent b4332a8
commit 49f1c1b
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git 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_intro.rst b/docs/source/api_ref_intro.rst
@@ -10,6 +10,7 @@ optimize your machine learning models.
    :maxdepth: 1
    :caption: Python API Reference
 
+   api_ref_sparsity
    api_ref_quantization
    api_ref_dtypes
    api_ref_kernel
diff --git a/docs/source/api_ref_sparsity.rst b/docs/source/api_ref_sparsity.rst
@@ -0,0 +1,18 @@
+.. _api_sparsity:
+
+================
+torchao.sparsity
+================
+.. automodule:: torchao.sparsity
+.. currentmodule:: torchao.sparsity
+
+.. autosummary::
+    :toctree: generated/
+    :nosignatures:
+
+    WandaSparsifier
+    PerChannelNormObserver
+    apply_sparse_semi_structured
+    apply_fake_sparsity
+
+
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -92,6 +92,7 @@ with more content coming soon.
    :maxdepth: 1
    :caption: API Reference
 
+   api_ref_sparsity
    api_ref_intro
    api_ref_quantization
    api_ref_dtypes