pytorch quantization ao migration phase 2: caffe2/benchmark (#65833)

vkuzo · facebook-github-bot · commit e3af4be96338 · 2021-10-01T06:17:36.000-07:00
Summary: Pull Request resolved: #65833 Renames `torch.quantization` to `torch.ao.quantization` in `caffe2/benchmarks` folder. ``` find caffe2/benchmarks/ -type f -name "*.py" -print0 | xargs -0 sed -i "s/torch\.quantization/torch.ao.quantization/g" ``` Test Plan: CI Reviewed By: z-a-f Differential Revision: D31275963 fbshipit-source-id: 8596bf28df5c3ad2c4490ac8abb285d6517c0116
diff --git a/benchmarks/operator_benchmark/pt/qobserver_test.py b/benchmarks/operator_benchmark/pt/qobserver_test.py
@@ -1,7 +1,7 @@
 
 import operator_benchmark as op_bench
 import torch
-import torch.quantization.observer as obs
+import torch.ao.quantization.observer as obs
 
 qobserver_short_configs_dict = {
     'attr_names': ('C', 'M', 'N', 'dtype', 'device'),
diff --git a/benchmarks/operator_benchmark/pt/qrnn_test.py b/benchmarks/operator_benchmark/pt/qrnn_test.py
@@ -41,9 +41,9 @@ def init(self, I, H, NL, B, D, dtype):
             bidirectional=D,
         )
         cell_temp = nn.Sequential(cell_nn)
-        self.cell = torch.quantization.quantize_dynamic(cell_temp,
-                                                        {nn.LSTM, nn.Linear},
-                                                        dtype=dtype)[0]
+        self.cell = torch.ao.quantization.quantize_dynamic(cell_temp,
+                                                           {nn.LSTM, nn.Linear},
+                                                           dtype=dtype)[0]
 
         x = torch.randn(sequence_len,  # sequence length
                         batch_size,    # batch size
diff --git a/benchmarks/operator_benchmark/pt/quantization_test.py b/benchmarks/operator_benchmark/pt/quantization_test.py
@@ -2,7 +2,7 @@
 import operator_benchmark as op_bench
 import torch
 import torch.nn.quantized as nnq
-import torch.quantization as tq
+import torch.ao.quantization as tq
 import torch.nn as nn
 
 """Microbenchmarks for general quantization operations."""
