Use solution_idx variable for hip_gemm_finalize

ROCm · Nov 7, 2024 · 4f8d530 · 4f8d530
1 parent 7180234
commit 4f8d530
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/targets/gpu/hip_gemm_impl.cpp b/src/targets/gpu/hip_gemm_impl.cpp
@@ -632,20 +632,19 @@ int32_t hip_gemm_finalize(context& ctx,
                           float beta,
                           int32_t solution_idx)
 {
-    int solution;
     auto gemm_item = hip_gemm_impl(output_shape, input_shapes, alpha, beta);
     if(solution_idx == 0)
     {
-        solution = gemm_item.tune(ctx, input_shapes);
+        solution_idx = gemm_item.tune(ctx, input_shapes);
         hip_gemm_save_solution(ctx, output_shape, input_shapes, solution);
     }
     // If a tuned solution index is already given, don't tune again but validate
-    // in case the data was tuned with a different hipBLAS version.
+    // in case the data was tuned with a different hipBLASLt version.
     else
     {
-        solution = gemm_item.validate(ctx, input_shapes, solution_idx);
+        solution_idx = gemm_item.validate(ctx, input_shapes, solution_idx);
     }
-    return solution;
+    return solution_idx;
 }
 
 int32_t hip_gemm_default_solution(context& ctx,