diff --git a/paddle/function/ConvOpTest.h b/paddle/function/ConvOpTest.h index ac55894c8de4102857aa317cd346c607e28a5c24..85debb7ae3a64bd25abd08a3c77a25b26ed80242 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") &&