[REFACTOR] Refactor test_quantize.py to use Gluon API (#20227)

* Refactor test_quantization to gluon API * review * Apply review suggestions * Skip flaky test
apache · Jun 17, 2021 · b54b7f3 · b54b7f3
1 parent 4f7f62c
commit b54b7f3
Show file tree

Hide file tree

Showing 2 changed files with 486 additions and 389 deletions.
diff --git a/src/operator/quantization/mkldnn/mkldnn_quantized_conv.cc b/src/operator/quantization/mkldnn/mkldnn_quantized_conv.cc
@@ -38,8 +38,6 @@ static void MKLDNNQuantizedConvForward(const nnvm::NodeAttrs& attrs,
                                        const std::vector<NDArray> &in_data,
                                        const std::vector<OpReqType> &req,
                                        const std::vector<NDArray> &out_data) {
-  CHECK_EQ(in_data[0].dtype(), mshadow::kUint8)
-    << "mkldnn_quantized_conv op only supports uint8 as input type";
   TmpMemMgr::Get()->Init(ctx.requested[conv::kTempSpace]);
   NDArray weight = in_data[conv::kWeight];
   ConvolutionParam param = nnvm::get<ConvolutionParam>(attrs.parsed);