From a6d1df3410eca93fecbe1acae5468889ee483b6a Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Tue, 7 May 2024 09:05:51 -0700 Subject: [PATCH 1/3] Fix build by removing the sparsity docs --- docs/source/api_ref_intro.rst | 1 - docs/source/api_ref_sparsity.rst | 14 -------------- docs/source/index.rst | 3 +-- 3 files changed, 1 insertion(+), 17 deletions(-) delete mode 100644 docs/source/api_ref_sparsity.rst diff --git a/docs/source/api_ref_intro.rst b/docs/source/api_ref_intro.rst index 8f7b84ef2a..c192ddceef 100644 --- a/docs/source/api_ref_intro.rst +++ b/docs/source/api_ref_intro.rst @@ -10,7 +10,6 @@ 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 deleted file mode 100644 index 9417d9befb..0000000000 --- a/docs/source/api_ref_sparsity.rst +++ /dev/null @@ -1,14 +0,0 @@ -.. _api_sparsity: - -================ -torchao.sparsity -================ - -.. currentmodule:: torchao.sparsity - -.. autosummary:: - :toctree: generated/ - :nosignatures: - - WandaSparsifier - PerChannelNormObserver diff --git a/docs/source/index.rst b/docs/source/index.rst index fb1649fa48..3ef94cd3fa 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -93,8 +93,7 @@ with more content coming soon. :caption: API Reference api_ref_intro - api_ref_sparsity api_ref_quantization api_ref_dtypes - .. +.. api_ref_kernel From b4332a887c66d3c22be46fd49f37fffed8bf8529 Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Tue, 7 May 2024 09:57:08 -0700 Subject: [PATCH 2/3] Trigger CI From 49f1c1bfb56259d98819cd26e749c0800d70e8a8 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Tue, 7 May 2024 10:03:08 -0700 Subject: [PATCH 3/3] Install dev requirements --- .github/workflows/doc_build.yml | 1 + docs/source/api_ref_intro.rst | 1 + docs/source/api_ref_sparsity.rst | 18 ++++++++++++++++++ docs/source/index.rst | 1 + 4 files changed, 21 insertions(+) create mode 100644 docs/source/api_ref_sparsity.rst diff --git a/.github/workflows/doc_build.yml b/.github/workflows/doc_build.yml index 2903e9cf81..125fa83e52 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_intro.rst b/docs/source/api_ref_intro.rst index c192ddceef..8f7b84ef2a 100644 --- 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 new file mode 100644 index 0000000000..8023d0bacc --- /dev/null +++ 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 index 3ef94cd3fa..cea80772d8 100644 --- 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