From ee4140103d4a7e56189492bb5fc9fd95d5eaf8d7 Mon Sep 17 00:00:00 2001
From: xzl <zlx_hg@163.com>
Date: Sat, 28 Oct 2017 13:36:01 +0800
Subject: [PATCH] add more dialtion test

---
 paddle/function/ConvOpTest.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/paddle/function/ConvOpTest.h b/paddle/function/ConvOpTest.h
index ac55894c8de..85debb7ae3a 100644
--- a/paddle/function/ConvOpTest.h
+++ b/paddle/function/ConvOpTest.h
@@ -79,9 +79,13 @@ void Convolution(const std::string& conv1,
             if (outputChannels < inputChannels) continue;
             for (size_t stride : {1, 2}) {
               for (size_t padding : {0, 1}) {
-                for (size_t dilation : {1}) {
+                for (size_t dilation : {1, 3}) {
                   if (padding >= filterSize) break;
 
+                  if ((conv1 == "NaiveConv-CPU" || conv2 == "NaiveConv-CPU") &&
+                      dilation > 1)
+                    break;
+
                   // NNPACK only supports stride = 1 if batchSize > 1
                   if ((conv1 == "NNPACKConv-CPU" ||
                        conv2 == "NNPACKConv-CPU") &&
-- 
GitLab