Skip to content

GitLab

Commit e29c7971 authored by Paul's avatar Paul
Browse files

Format

parent e53de5b6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/targets/gpu/gemm_impl.cpp
View file @ e29c7971
...@@ -541,14 +541,14 @@ int32_t gemm_finalize(context& ctx, ...@@ -541,14 +541,14 @@ int32_t gemm_finalize(context& ctx,
if(solution_idx == 0) if(solution_idx == 0)
{ {
auto gemm_item = gemm_impl<float>(output_shape, input_shapes, alpha, beta, compute_fp32); auto gemm_item = gemm_impl<float>(output_shape, input_shapes, alpha, beta, compute_fp32);
solution_idx = gemm_item.tune(ctx, input_shapes); solution_idx = gemm_item.tune(ctx, input_shapes);
} }
else else
{ {
// If a tuned solution index is already given, don't tune again but validate // If a tuned solution index is already given, don't tune again but validate
// in case the data was tuned with a different rocBLAS version // in case the data was tuned with a different rocBLAS version
auto gemm_item = gemm_impl<float>(output_shape, input_shapes, alpha, beta, compute_fp32); auto gemm_item = gemm_impl<float>(output_shape, input_shapes, alpha, beta, compute_fp32);
solution_idx = gemm_item.validate(ctx, input_shapes, solution_idx); solution_idx = gemm_item.validate(ctx, input_shapes, solution_idx);
} }
#else #else
(void)ctx, (void)output_shape, (void)input_shapes; (void)ctx, (void)output_shape, (void)input_shapes;
...@@ -573,14 +573,14 @@ int32_t gemm_finalize(context& ctx, ...@@ -573,14 +573,14 @@ int32_t gemm_finalize(context& ctx,
if(solution_idx == 0) if(solution_idx == 0)
{ {
auto gemm_item = gemm_impl<int32_t>(output_shape, input_shapes, alpha, beta, compute_fp32); auto gemm_item = gemm_impl<int32_t>(output_shape, input_shapes, alpha, beta, compute_fp32);
solution_idx = gemm_item.tune(ctx, input_shapes); solution_idx = gemm_item.tune(ctx, input_shapes);
} }
else else
{ {
// If a tuned solution index is already given, don't tune again but validate // If a tuned solution index is already given, don't tune again but validate
// in case the data was tuned with a different rocBLAS version // in case the data was tuned with a different rocBLAS version
auto gemm_item = gemm_impl<int32_t>(output_shape, input_shapes, alpha, beta, compute_fp32); auto gemm_item = gemm_impl<int32_t>(output_shape, input_shapes, alpha, beta, compute_fp32);
solution_idx = gemm_item.validate(ctx, input_shapes, solution_idx); solution_idx = gemm_item.validate(ctx, input_shapes, solution_idx);
} }
#else #else
(void)ctx, (void)output_shape, (void)input_shapes; (void)ctx, (void)output_shape, (void)input_shapes;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment