add test for diff padding mode

src/eliminate_pad.cpp

@@ -41,6 +41,8 @@ void eliminate_pad::update_op(T,
     std::array<size_t, 2> new_pads{static_cast<size_t>(pads[2]), static_cast<size_t>(pads[3])};
 
     T op       = any_cast<T>(ins->get_operator());
+    if(op.padding_mode != op::padding_mode_t::default_)
+        return;
     op.padding = new_pads;
 
     std::vector<instruction_ref> new_inputs{ins->inputs()};

test/eliminate_pad_test.cpp

@@ -27,13 +27,15 @@ create_im2col(migraphx::instruction_ref& l_img, size_t channels, migraphx::progr
 }
 
 migraphx::instruction_ref
-create_conv(migraphx::instruction_ref& l_img, size_t channels, migraphx::program& p)
+create_conv(migraphx::instruction_ref& l_img, size_t channels, migraphx::program& p, migraphx::op::padding_mode_t padding_mode = migraphx::op::padding_mode_t::default_)
 {
     migraphx::shape s_weights{migraphx::shape::int32_type, {4, channels, 3, 3}};
     std::vector<int32_t> weights(4 * channels * 3 * 3);
 
     auto l_weights = p.add_literal(migraphx::literal{s_weights, weights});
-    return p.add_instruction(migraphx::op::convolution{}, l_img, l_weights);
+    migraphx::op::convolution op;
+    op.padding_mode = padding_mode;
+    return p.add_instruction(op, l_img, l_weights);
 }
 
 TEST_CASE(rewrite_test)
@@ -78,4 +80,23 @@ TEST_CASE(rewrite_test_asymmetric)
         p.begin(), p.end(), [](const migraphx::instruction& ins) { return ins.name() == "pad"; }));
 }
 
+TEST_CASE(rewrite_test_same_padding)
+{
+    migraphx::program p;
+    size_t img_dim[2] = {2, 2};
+    size_t channels   = 1;
+    std::vector<int32_t> input(channels * img_dim[0] * img_dim[1]);
+    std::iota(input.begin(), input.end(), 0);
+
+    migraphx::shape s_img{migraphx::shape::int32_type, {1, channels, img_dim[0], img_dim[1]}};
+    auto l_img      = p.add_literal(migraphx::literal{s_img, input});
+    auto padded_img = p.add_instruction(migraphx::op::pad{{0, 0, 1, 1, 0, 0, 1, 1}}, l_img);
+    
+    create_conv(padded_img, channels, p, migraphx::op::padding_mode_t::same);
+
+    p.compile(eliminate_pad_target{});
+    EXPECT(std::any_of(
+        p.begin(), p.end(), [](const migraphx::instruction& ins) { return ins.name() == "pad"; }));
+}
+
 int main(int argc, const char* argv[]) { test::run(argc, argv); }