only return one solution

src/targets/gpu/convolution.cpp

@@ -120,35 +120,24 @@ shape miopen_convolution::find(context& ctx, const shape& output_shape, std::vec
     if(status != miopenStatusSuccess)
         MIGRAPHX_THROW("MIOpen Convolution: find convolution failed");
 
-    size_t solution_count;
-
-    status = miopenConvolutionForwardGetSolutionCount(ctx.get_stream().get_miopen(),
-                                                      w_desc.get(),
-                                                      x_desc.get(),
-                                                      cd.get(),
-                                                      y_desc.get(),
-                                                      &solution_count);
-    if(status != miopenStatusSuccess)
-        MIGRAPHX_THROW("MIOpen Convolution: get solution count failed");
-
-    std::vector<miopenConvSolution_t> solutions(solution_count);
+    size_t solution_count = 1;
+    miopenConvSolution_t conv_solution;
 
     status = miopenConvolutionForwardGetSolution(ctx.get_stream().get_miopen(),
                                                  w_desc.get(),
                                                  x_desc.get(),
                                                  cd.get(),
                                                  y_desc.get(),
-                                                 solution_count,
+                                                 1,
                                                  &solution_count,
-                                                 solutions.data());
-    if(status != miopenStatusSuccess)
+                                                 &conv_solution);
+    if(status != miopenStatusSuccess or solution_count != 1)
         MIGRAPHX_THROW("MIOpen Convolution: get solution failed");
 
-    const auto& best_solution = solutions.front();
-    solution_id               = best_solution.solution_id;
-    algo                      = best_solution.algorithm;
+    solution_id               = conv_solution.solution_id;
+    algo                      = conv_solution.algorithm;
 
-    return shape{shape::int8_type, {best_solution.workspace_size}};
+    return shape{shape::int8_type, {conv_solution.workspace_size}};
 }
 
 void miopen_convolution::finalize(context& ctx,