Update on "[Quantization] Add metadata porting for nodes added by quantization"

kimishpatel · kimishpatel · commit 73f9855c25c0 · 2023-09-01T13:19:18.000-07:00
Summary: This diff adds adding metadata to q-dq nodes by inferring the quatization intent from node annotations. Annotations on the node are way for user to specify how a node or subgraph is supposed to be quantized. We continue to use that information to copy metadata on Q/DQ node from appropriate nodes. Test Plan: Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D48488416](https://our.internmc.facebook.com/intern/diff/D48488416) [ghstack-poisoned]
diff --git a/test/quantization/pt2e/test_metadata_porting.py b/test/quantization/pt2e/test_metadata_porting.py
@@ -1,5 +1,7 @@
 # Owner(s): ["oncall: quantization"]
 import copy
+
+import unittest
 from typing import List
 
 import torch
@@ -15,8 +17,6 @@
 
 from torch.testing._internal.common_quantization import QuantizationTestCase
 
-import unittest
-
 
 class TestHelperModules:
     class Conv2dWithObsSharingOps(torch.nn.Module):
