fix user_defined quant unittest (#27385)

7dbc5126 · Bai Yifan · GitHub · 5216d055 · 7dbc5126
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py ...luid/contrib/slim/tests/test_user_defined_quantization.py +1 -0

未找到文件。
--- 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)