frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[quant][pt2e] Add generate_numeric_debug_handle pass (#114315)

Browse Source 
Summary:
This is a util for numeric suite in pt2 export so that we can build
a more streamlined UX for numerical debugging in quant + executorch stack

Test Plan:
python test/test_quantization.py TestGenerateNumericDebugHandle

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/114315
Approved by: https://github.com/zhxchen17
This commit is contained in:
Jerry Zhang
2023-11-30 15:33:33 -08:00
committed by PyTorch MergeBot
parent 2dd2fb91d9
commit 64fd706b21
7 changed files with 56 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_quantization.py
View File

@ -85,6 +85,7 @@ try:
from quantization.pt2e.test_graph_utils import TestGraphUtils # noqa: F401
from quantization.pt2e.test_duplicate_dq import TestDuplicateDQPass # noqa: F401
from quantization.pt2e.test_metadata_porting import TestMetaDataPorting # noqa: F401
from quantization.pt2e.test_generate_numeric_debug_handle import TestGenerateNumericDebugHandle # noqa: F401
from quantization.pt2e.test_quantize_pt2e import TestQuantizePT2E # noqa: F401
from quantization.pt2e.test_representation import TestPT2ERepresentation # noqa: F401
from quantization.pt2e.test_xnnpack_quantizer import TestXNNPACKQuantizer # noqa: F401