mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
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:
committed by
PyTorch MergeBot
parent
535016967a
commit
b893aa71ca
@ -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
|
||||
|
@ -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()
|
@ -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
|
||||
|
@ -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
|
||||
|
@ -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__ = [
|
||||
|
@ -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__ = [
|
||||
|
Reference in New Issue
Block a user