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
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 = 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>>;
type init = lowest();
......
......@@ -83,12 +83,12 @@ void gemm_impl(context& ctx,
auto a_lens = args[0].get_shape().lens();
auto b_lens = args[1].get_shape().lens();
output_shape.visit_type([&](auto as) {
auto alpha_r = as(alpha);
auto beta_r = as(beta);
auto alpha_r = as(alpha);
auto beta_r = as(beta);
if(compute_fp32)
{
alpha_r = alpha;
beta_r = beta;
alpha_r = alpha;
beta_r = beta;
}
auto out_lens = output_shape.lens();
rocblas_int m = out_lens[dim_0];
......
......@@ -85,7 +85,13 @@ struct rocblas_gemm
}
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();
}
......
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