Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 521fd75

Browse files
committed
enable int8 bn test only in mkldnn backend
1 parent 8c56e20 commit 521fd75

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/python/quantization/test_quantization.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -606,6 +606,16 @@ def get_mean_var(data):
606606
return mean, var
607607

608608
def check_quantized_bn(data_shape, qdtype):
609+
if qdtype == 'uint8':
610+
print('skipped testing quantize_bn for uint8 since it is not supported yet')
611+
return
612+
elif is_test_for_native_cpu():
613+
print('skipped testing quantize_bn for native cpu since it is not supported yet')
614+
return
615+
elif is_test_for_gpu():
616+
print('skipped testing quantize_bn for gpu since it is not supported yet')
617+
return
618+
609619
# qdtype = int8
610620
data_low = -127.0
611621
data_high = 127.0

0 commit comments

Comments
 (0)