Update on "[quant][graphmode] Quantization support for quantized::add_scalar_relu and quantized::add_scalar_relu_out"

jerryzh168 · jerryzh168 · commit a6b30a3ba5fb · 2020-03-30T12:00:54.000-07:00
Summary: Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20742138](https://our.internmc.facebook.com/intern/diff/D20742138) [ghstack-poisoned]
diff --git a/test/test_jit.py b/test/test_jit.py
@@ -1609,6 +1609,8 @@ def forward(self, x, y):
             FileCheck().check_not("aten::add") \
                        .check_not("aten::relu") \
                        .check_not("aten::relu_") \
+                       .check_not("quantized::add") \
+                       .check_not("quantized::relu") \
                        .check("quantized::add_relu") \
                        .run(m.graph_for(data, data))
 
@@ -1639,6 +1641,8 @@ def forward(self, x):
             FileCheck().check_not("aten::add") \
                        .check_not("aten::add_") \
                        .check_not("aten::relu") \
+                       .check_not("quantized::add_scalar") \
+                       .check_not("quantized::relu") \
                        .check("quantized::add_scalar_relu") \
                        .run(m.graph_for(data))
 
