Commit 8d21ccdf authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent 3df20646
...@@ -22,7 +22,7 @@ void softmax(hipStream_t stream, const argument& result, const argument& arg, in ...@@ -22,7 +22,7 @@ void softmax(hipStream_t stream, const argument& result, const argument& arg, in
hip_visit_all(result, arg, batch_shape)([&](auto output, auto input, auto batch) { hip_visit_all(result, arg, batch_shape)([&](auto output, auto input, auto batch) {
const index_int max_block_size = 120; const index_int max_block_size = 120;
// const index_int max_block_size = 128; // const index_int max_block_size = 128;
const index_int block_size = compute_block_size(batch_item_num, max_block_size); const index_int block_size = compute_block_size(batch_item_num, max_block_size);
using type = device_type<std::remove_cv_t<typename decltype(input)::value_type>>; using type = device_type<std::remove_cv_t<typename decltype(input)::value_type>>;
type init = lowest(); type init = lowest();
......
...@@ -83,12 +83,12 @@ void gemm_impl(context& ctx, ...@@ -83,12 +83,12 @@ 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 = as(alpha);
auto beta_r = as(beta); auto beta_r = as(beta);
if(compute_fp32) if(compute_fp32)
{ {
alpha_r = alpha; alpha_r = alpha;
beta_r = beta; 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];
......
...@@ -85,7 +85,13 @@ struct rocblas_gemm ...@@ -85,7 +85,13 @@ struct rocblas_gemm
} }
else else
{ {
gemm(ctx, output_shape, args, int32_t(alpha), int32_t(beta), int8_x4_format, compute_fp32); gemm(ctx,
output_shape,
args,
int32_t(alpha),
int32_t(beta),
int8_x4_format,
compute_fp32);
} }
return args.back(); return args.back();
} }
......
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