From 4fda46bd8c0e9f68376a0b6c69f8027ab227f650 Mon Sep 17 00:00:00 2001 From: Xinyu Chen Date: Fri, 27 Dec 2019 08:45:19 +0800 Subject: [PATCH] fix py27 quantization (#17153) --- python/mxnet/contrib/quantization.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/mxnet/contrib/quantization.py b/python/mxnet/contrib/quantization.py index a56f8531aa49..ce22fb753ace 100644 --- a/python/mxnet/contrib/quantization.py +++ b/python/mxnet/contrib/quantization.py @@ -613,7 +613,9 @@ def quantize_model_mkldnn(sym, arg_params, aux_params, A tuple of quantized symbol, quantized arg_params, and aux_params. ------- """ - if ctx != cpu(): + if not isinstance(ctx, Context): + raise ValueError('currently only supports single ctx, while received %s' % str(ctx)) + if ctx.device_type != 'cpu': raise ValueError( 'quantize_model_mkldnn only support Intel cpu platform with MKL-DNN Backend')