Skip to content

Commit 1801fc3

Browse files
authored
[quant][pt2e] Add documentation for pt2 export static quant (#2512)
Add pt2 export docs Renames the current pt2e tutorial
1 parent c32ad93 commit 1801fc3

File tree

4 files changed

+479
-126
lines changed

4 files changed

+479
-126
lines changed

prototype_source/fx_graph_mode_ptq_static.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -241,8 +241,8 @@ of the observers for activation and weight. ``QConfigMapping`` contains mapping
241241
242242
qconfig_mapping = (QConfigMapping()
243243
.set_global(qconfig_opt) # qconfig_opt is an optional qconfig, either a valid qconfig or None
244-
.set_object_type(torch.nn.Conv2d, qconfig_opt) # can be a callable...
245-
.set_object_type("torch.nn.functional.add", qconfig_opt) # ...or a string of the class name
244+
.set_object_type(torch.nn.Conv2d, qconfig_opt) # can be a callable...
245+
.set_object_type("reshape", qconfig_opt) # ...or a string of the method
246246
.set_module_name_regex("foo.*bar.*conv[0-9]+", qconfig_opt) # matched in order, first match takes precedence
247247
.set_module_name("foo.bar", qconfig_opt)
248248
.set_module_name_object_type_order()

prototype_source/prototype_index.rst

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,12 +69,19 @@ Prototype features are not available as part of binary distributions like PyPI o
6969
:tags: Debugging,Quantization
7070

7171
.. customcarditem::
72-
:header: Quantization in PyTorch 2.0 Export Tutorial
73-
:card_description: Learn how to use the Quantization in PyTorch 2.0 Export.
72+
:header: How to Write a Quantizer for PyTorch 2.0 Export Quantization
73+
:card_description: Learn how to implement a Quantizer for PT2.0 Export Quantization
7474
:image: ../_static/img/thumbnails/cropped/generic-pytorch-logo.png
75-
:link: ../prototype/quantization_in_pytorch_2_0_export_tutorial.html
75+
:link: ../prototype/pt2e_quantizer.html
7676
:tags: Quantization
7777

78+
.. customcarditem::
79+
:header: PyTorch 2.0 Export Post Training Static Quantization
80+
:card_description: Learn how to use Post Training Static Quantization in PyTorch 2.0 Export.
81+
:image: ../_static/img/thumbnails/cropped/generic-pytorch-logo.png
82+
:link: ../prototype/pt2e_quant_ptq_static.html
83+
:tags: Quantization
84+
7885
.. Mobile
7986
8087
.. customcarditem::

0 commit comments

Comments
 (0)