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 b83b40b86b2ddbdd2e8da0fd513c45b54d8e88ca..b484a88273b20ff2f14a4e2f1cd5554615453ff6 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(