From f6ac5990aabbde8925b74c1ed6fd7caf34c2179e Mon Sep 17 00:00:00 2001
From: yukavio <67678385+yukavio@users.noreply.github.com>
Date: Fri, 31 Jul 2020 14:53:48 +0800
Subject: [PATCH] fix quant unit test (#25792)

---
 .../fluid/contrib/slim/tests/test_user_defined_quantization.py   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py b/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
index 0c8f5cdd84c..c9ea15bf6cd 100644
--- a/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
@@ -47,6 +47,7 @@ def residual_block(img, label, num=1):
             num_filters=ch_out,
             stride=stride,
             padding=padding,
+            use_cudnn=False,
             act=None,
             bias_attr=bias_attr)
         return fluid.layers.batch_norm(input=tmp, act=act)
-- 
GitLab