fix user_defined quant unittest (#27385)

5 years ago · 7dbc5126ba
parent 5216d05510
commit 7dbc5126ba
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
@ -48,6 +48,7 @@ def residual_block(img, label, num=1):
            num_filters=ch_out,
            stride=stride,
            padding=padding,
+            use_cudnn=use_cudnn,
            act=None,
            bias_attr=bias_attr)
        return fluid.layers.batch_norm(input=tmp, act=act)