From 9bc223c8a293bef13fb24c38107034ea81047ad5 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Tue, 7 Apr 2020 10:22:23 +0800 Subject: [PATCH] fix test_graph_wrapper failure on cudnnv7, test=develop (#23451) --- python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py | 1 + 1 file changed, 1 insertion(+) 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 5340f36196..430ac569f6 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) -- GitLab