diff --git a/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py b/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py
index 5340f36196edc9c011a98faee5fbf5a6fbc7a639..430ac569f601d09ded9dc90df8b7cdd0d2b4b5a5 100644
--- a/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py
@@ -37,6 +37,7 @@ def residual_block(num):
             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)