Rename generate_numeric_debug_handle to numeric_debugger (#130590)

Summary:
att

Test Plan:
CI

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/130590
Approved by: https://github.com/dulinriley, https://github.com/tarun292
This commit is contained in:
Jerry Zhang
2024-07-15 21:40:24 +00:00
committed by PyTorch MergeBot
parent 535016967a
commit b893aa71ca
7 changed files with 7 additions and 7 deletions

View File

@ -635,4 +635,4 @@ the `custom operator mechanism <https://pytorch.org/tutorials/advanced/torch_scr
.. automodule:: torch.nn.quantized.dynamic.modules
.. automodule:: torch.quantization
.. automodule:: torch.nn.intrinsic.modules
.. automodule:: torch.ao.quantization.pt2e.generate_numeric_debug_handle
.. automodule:: torch.ao.quantization.pt2e.numeric_debugger

View File

@ -35,7 +35,7 @@ def is_fbcode():
@unittest.skipIf(IS_WINDOWS, "Windows not yet supported for torch.compile")
class TestGenerateNumericDebugHandle(TestCase):
class TestNumericDebugger(TestCase):
def test_simple(self):
m = TestHelperModules.Conv2dThenConv1d()
example_inputs = m.example_inputs()

View File

@ -85,7 +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_numeric_debugger import TestNumericDebugger # 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

View File

@ -16,8 +16,8 @@ from .stubs import * # noqa: F403
from .pt2e.export_utils import _move_exported_model_to_eval as move_exported_model_to_eval
from .pt2e.export_utils import _move_exported_model_to_train as move_exported_model_to_train
from .pt2e.export_utils import _allow_exported_model_train_eval as allow_exported_model_train_eval
from .pt2e.generate_numeric_debug_handle import generate_numeric_debug_handle # noqa: F401
from .pt2e.generate_numeric_debug_handle import NUMERIC_DEBUG_HANDLE_KEY # noqa: F401
from .pt2e.numeric_debugger import generate_numeric_debug_handle # noqa: F401
from .pt2e.numeric_debugger import NUMERIC_DEBUG_HANDLE_KEY # noqa: F401
from typing import Union, List, Callable, Tuple, Optional
from torch import Tensor
import torch

View File

@ -75,7 +75,7 @@ from .custom_config import (
from .lower_to_fbgemm import lower_to_fbgemm
# importing the lib so that the quantized_decomposed ops are registered
from ._decomposed import quantized_decomposed_lib # noqa: F401
from torch.ao.quantization.pt2e.generate_numeric_debug_handle import NUMERIC_DEBUG_HANDLE_KEY
from torch.ao.quantization.pt2e.numeric_debugger import NUMERIC_DEBUG_HANDLE_KEY
import operator
__all__ = [

View File

@ -24,7 +24,7 @@ from torch.ao.quantization.quantizer import (
QuantizationSpecBase,
)
from torch.ao.quantization import ObserverOrFakeQuantize
from torch.ao.quantization.pt2e.generate_numeric_debug_handle import NUMERIC_DEBUG_HANDLE_KEY
from torch.ao.quantization.pt2e.numeric_debugger import NUMERIC_DEBUG_HANDLE_KEY
# TODO: make pt2e folder private?
__all__ = [