enable int8 bn test only in mkldnn backend

apache · Jul 29, 2019 · 521fd75 · 521fd75
1 parent 8c56e20
commit 521fd75
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/python/quantization/test_quantization.py b/tests/python/quantization/test_quantization.py
@@ -606,6 +606,16 @@ def get_mean_var(data):
         return mean, var
 
     def check_quantized_bn(data_shape, qdtype):
+        if qdtype == 'uint8':
+            print('skipped testing quantize_bn for uint8 since it is not supported yet')
+            return
+        elif is_test_for_native_cpu():
+            print('skipped testing quantize_bn for native cpu since it is not supported yet')
+            return
+        elif is_test_for_gpu():
+            print('skipped testing quantize_bn for gpu since it is not supported yet')
+            return
+
         # qdtype = int8
         data_low = -127.0
         data_high = 127.0