Add to conv dynamic batch test

src/program.cpp

@@ -311,6 +311,8 @@ std::vector<argument> generic_eval(const module* mod,
                     });
                     return shapes;
                 };
+                // TODO: Consider how this will be handled when memoized.
+                // Could memoize these output shapes now so not recalculating
                 output_shape = ins->get_operator().compute_shape(to_shapes(values));
             }
             else
@@ -332,7 +334,10 @@ std::vector<argument> generic_eval(const module* mod,
         }
         assert(results.find(ins) != results.end());
         // TODO: update this assert for dynamic shapes
-        // assert(results.at(ins).get_shape() == ins->get_shape());
+        if(not ins->get_shape().dynamic())
+        {
+            assert(results.at(ins).get_shape() == ins->get_shape());
+        }
     }
     return {results.at(std::prev(mod->end()))};
 }

test/ref_ops_test.cpp

@@ -910,17 +910,61 @@ TEST_CASE(conv_dynamic_batch_test)
                               -0.16138598,
                               0.79344082};
 
-    migraphx::shape input_fixed_shape{migraphx::shape::float_type, {2, 3, 4, 4}};
+    migraphx::shape input_fixed_shape0{migraphx::shape::float_type, {2, 3, 4, 4}};
 
-    migraphx::parameter_map params;
-    params["X"] = migraphx::argument(input_fixed_shape, a.data());
-    params["W"] = migraphx::argument(weights_shape, c.data());
+    migraphx::parameter_map params0;
+    params0["X"] = migraphx::argument(input_fixed_shape0, a.data());
+    params0["W"] = migraphx::argument(weights_shape, c.data());
 
-    auto result = p.eval(params).back();
+    auto result = p.eval(params0).back();
     std::vector<float> results_vector(64);
     result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
 
     EXPECT(migraphx::verify_range(results_vector, sol));
+
+    a = {2.71567607,  -0.9960829,  0.91671127,  0.28140706,  0.63235772,  0.08077253,  0.80927712,
+         -0.59108931, -1.05421555, -2.76622486, -0.85044265, -0.52049929, 0.67726439,  -0.65290606,
+         0.02345525,  -0.33579525, 0.38901961,  1.05473483,  -1.31188095, 1.8963089,   -0.07265259,
+         0.947339,    0.41949373,  -0.70814759, 0.25892952,  1.07311416,  1.2571274,   -0.62318051,
+         -0.19951548, -0.94232577, -0.29393643, 0.42292568,  -0.80230367, 1.40909171,  0.63617158,
+         0.13900366,  1.09253144,  -0.15265895, 1.54781747,  0.72780299,  1.09189606,  -0.38068101,
+         0.97057933,  -0.58958799, 1.56188643,  0.21474874,  0.58725154,  -1.27097559, -0.03024297,
+         1.09437096,  -0.4897908,  0.34838957,  -1.31042492, -1.69069934, 0.86956722,  -0.40457946,
+         0.46691212,  1.29273605,  0.26464137,  0.22073045,  -1.02178168, 0.22163901,  -1.84387338,
+         0.75522131,  -0.45775682, -0.42241111, -1.50944722, 1.07256448,  -1.95876884, -0.28106022,
+         0.3341668,   2.13129425,  -1.14728117, -1.06555498, -0.298444,   -0.88322699, -0.65866792,
+         -2.06007552, 0.01374334,  0.45612028,  0.52715492,  1.01914406,  -1.72659791, 0.80650896,
+         0.16860051,  2.24112225,  -0.78620857, 0.36566174,  -0.07020134, -0.47976932, -0.68230027,
+         -0.94711417, -0.54506505, 1.66504931,  -0.71860826, 0.61132306};
+
+    c = {-0.14601797, -0.13000923, 0.06521662,  0.06178288,  -0.11083675, 0.10154136,  0.09990512,
+         0.06030385,  -0.11374587, -0.17523311, -0.14344215, 0.17802463,  0.06300922,  -0.15325832,
+         0.07066704,  0.05166031,  0.00615084,  -0.02606523, 0.08083995,  -0.17913306, 0.0624622,
+         0.0735731,   -0.04198661, -0.0164391,  -0.06374192, 0.16569914,  0.10681538,  0.07370754,
+         0.02802075,  0.00282027,  0.15104802,  -0.11084409, -0.00197773, 0.07924436,  0.03528272,
+         0.04765259,  -0.15896152, 0.07917164,  0.12125669,  -0.1154705,  -0.11999125, 0.12749968,
+         -0.06269585, 0.18658121,  -0.03944227, 0.0111798,   -0.17731084, 0.11789055,  -0.09982193,
+         0.08142821,  0.0729029,   0.11303909,  0.12735154,  0.03885292};
+
+    sol = {-0.20817225,
+           0.87965256,
+           0.14958936,
+           -1.24887264,
+           -0.06540672,
+           0.20778663,
+           0.40456355,
+           -0.99900877};
+
+    migraphx::shape input_fixed_shape1{migraphx::shape::float_type, {1, 3, 4, 4}};
+
+    migraphx::parameter_map params1;
+    params1["X"] = migraphx::argument(input_fixed_shape1, a.data());
+    params1["W"] = migraphx::argument(weights_shape, c.data());
+
+    result = p.eval(params1).back();
+    result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
+
+    EXPECT(migraphx::verify_range(results_vector, sol));
 }
 
 TEST_CASE(conv2d_padding_stride_test)