fixing imports for moved test

Summary:

Test Plan: python test.py

Reviewers:

Subscribers:

Tasks:

Tags:

ghstack-source-id: 7d4cabb
Pull Request resolved: #6

Author: HDCharles

test/test.py

@@ -14,15 +14,15 @@
 
 from torch.ao.quantization import MinMaxObserver, QConfigMapping
 
-from dynamic_quant import (
+from torchao.quantization.dynamic_quant import (
     DynamicallyPerAxisQuantizedLinear,
 )
-from quant_api import (
+from torchao.quantization.quant_api import (
     apply_dynamic_quant,
     apply_weight_only_int8_quant,
     change_linear_weights_to_dqtensors,
 )
-from quant_primitives import (
+from torchao.quantization.quant_primitives import (
     dequantize_per_channel,
     dequantize_per_tensor,
     dynamically_quantize_per_channel,
@@ -33,17 +33,17 @@
     safe_int_mm,
 )
 
-from smoothquant import (
+from torchao.quantization.smoothquant import (
     get_scale,
     replace_with_custom_fn_if_matches_filter,
     smooth_fq_linear_to_inference,
     SmoothFakeDynamicallyQuantizedLinear,
     swap_linear_with_smooth_fq_linear,
 )
-from subclass import (
+from torchao.quantization.subclass import (
     DynamicallyQuantizedLinearWeight,
 )
-from utils import (
+from torchao.quantization.utils import (
     apply_logging_hook,
     compute_error,
     compute_error as SQNR,

torchao/quantization/__init__.py

@@ -4,12 +4,12 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 
-from smoothquant import *  # noqa: F403
-from quant_api import *  # noqa: F403
-from subclass import *  # noqa: F403
-from quant_primitives import *  # noqa: F403
-from utils import *  # noqa: F403
-from weight_only import *  # noqa: F403
+from .smoothquant import *  # noqa: F403
+from .quant_api import *  # noqa: F403
+from .subclass import *  # noqa: F403
+from .quant_primitives import *  # noqa: F403
+from .utils import *  # noqa: F403
+from .weight_only import *  # noqa: F403
 
 __all__ = [
     "DynamicallyPerAxisQuantizedLinear",

torchao/quantization/dynamic_quant.py

@@ -6,7 +6,7 @@
 
 import torch
 import torch.nn as nn
-from quant_primitives import (
+from .quant_primitives import (
     dynamically_quantize_per_channel,
     quant_int8_dynamic_per_token_linear,
 )

torchao/quantization/quant_api.py

@@ -12,13 +12,13 @@
 """
 
 import torch
-from dynamic_quant import (
+from .dynamic_quant import (
     DynamicallyPerAxisQuantizedLinear,
 )
-from subclass import (
+from .subclass import (
     DynamicallyQuantizedLinearWeight,
 )
-from weight_only import (
+from .weight_only import (
     WeightOnlyInt8QuantLinear,
 )
 

torchao/quantization/smoothquant.py

@@ -14,9 +14,9 @@
 
 import torch
 import torch.nn.functional as F
-import quant_api
+import torchao.quantization.quant_api as quant_api
 
-from quant_primitives import (
+from .quant_primitives import (
     dynamically_quantize_per_channel,
     quant_int8_dynamic_per_token_linear,
 )

torchao/quantization/subclass.py

@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 import torch
-from quant_primitives import (
+from .quant_primitives import (
     dequantize_per_channel,
     dynamically_quantize_per_channel,
     quant_int8_dynamic_per_token_linear,

torchao/quantization/weight_only.py

@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 import torch
-from quant_primitives import (
+from .quant_primitives import (
     dynamically_quantize_per_channel,
 )