Commit adcf522c authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

change type for alpha beta

parent 9f755219
...@@ -83,13 +83,9 @@ void gemm_impl(context& ctx, ...@@ -83,13 +83,9 @@ void gemm_impl(context& ctx,
auto a_lens = args[0].get_shape().lens(); auto a_lens = args[0].get_shape().lens();
auto b_lens = args[1].get_shape().lens(); auto b_lens = args[1].get_shape().lens();
output_shape.visit_type([&](auto as) { output_shape.visit_type([&](auto as) {
auto alpha_r = as(alpha); auto alpha_r = compute_fp32 ? alpha : as(alpha);
auto beta_r = as(beta); auto beta_r = compute_fp32 ? beta : as(beta);
if(compute_fp32)
{
alpha_r = alpha;
beta_r = beta;
}
auto out_lens = output_shape.lens(); auto out_lens = output_shape.lens();
rocblas_int m = out_lens[dim_0]; rocblas_int m = out_lens[dim_0];
rocblas_int n = out_lens[dim_1]; rocblas_int n = out_lens[dim_1];
......
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