From b28c374ad29e00a267c736945a98066bc65b2089 Mon Sep 17 00:00:00 2001 From: feng_shuai Date: Wed, 15 Dec 2021 13:09:26 +0800 Subject: [PATCH] reduce the pass run-time (#38115) --- .../ir/inference/test_mkldnn_depthwise_conv_pass.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_mkldnn_depthwise_conv_pass.py b/python/paddle/fluid/tests/unittests/ir/inference/test_mkldnn_depthwise_conv_pass.py index b83b40b86b2..b484a88273b 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/test_mkldnn_depthwise_conv_pass.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/test_mkldnn_depthwise_conv_pass.py @@ -41,10 +41,10 @@ class DepthwiseConvMKLDNNPass(PassAutoScanTest): def sample_program_config(self, draw): # generate random number random_batch_size = draw(st.integers(min_value=1, max_value=4)) - random_channel = draw(st.integers(min_value=2, max_value=64)) - random_input_dim1 = draw(st.integers(min_value=50, max_value=512)) - random_input_dim2 = draw(st.integers(min_value=50, max_value=512)) - random_out_channel = draw(st.integers(min_value=20, max_value=256)) + random_channel = draw(st.integers(min_value=2, max_value=10)) + random_input_dim1 = draw(st.integers(min_value=20, max_value=50)) + random_input_dim2 = draw(st.integers(min_value=20, max_value=50)) + random_out_channel = draw(st.integers(min_value=20, max_value=25)) random_groups = draw(st.integers(min_value=1, max_value=3)) random_dilations = draw( -- GitLab