Commit 645fe9d6 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 4e028da0
...@@ -180,8 +180,8 @@ struct miopen_apply ...@@ -180,8 +180,8 @@ struct miopen_apply
auto&& op = any_cast<op::quant_dot>(ins->get_operator()); auto&& op = any_cast<op::quant_dot>(ins->get_operator());
auto inputs = ins->inputs(); auto inputs = ins->inputs();
auto in_shapes = to_shapes(inputs); auto in_shapes = to_shapes(inputs);
auto pack_a = insert_allocation(ins, in_shapes[0], "pack_a"); auto pack_a = insert_allocation(ins, in_shapes[0], "pack_a");
auto pack_b = insert_allocation(ins, in_shapes[1], "pack_b"); auto pack_b = insert_allocation(ins, in_shapes[1], "pack_b");
auto output = insert_allocation(ins, ins->get_shape()); auto output = insert_allocation(ins, ins->get_shape());
inputs.push_back(pack_a); inputs.push_back(pack_a);
inputs.push_back(pack_b); inputs.push_back(pack_b);
......
...@@ -70,7 +70,7 @@ argument miopen_quant_gemm::compute(context& ctx, ...@@ -70,7 +70,7 @@ argument miopen_quant_gemm::compute(context& ctx,
auto n_dim = output_shape.lens().size(); auto n_dim = output_shape.lens().size();
auto dim_1 = n_dim - 1; auto dim_1 = n_dim - 1;
auto dim_0 = n_dim - 2; auto dim_0 = n_dim - 2;
auto arg_num = args.size(); auto arg_num = args.size();
rocblas_int lda = args[0].get_shape().strides()[transa ? dim_1 : dim_0]; rocblas_int lda = args[0].get_shape().strides()[transa ? dim_1 : dim_0];
rocblas_int ldb = args[1].get_shape().strides()[transb ? dim_1 : dim_0]; rocblas_int ldb = args[1].get_shape().strides()[transb ? dim_1 : dim_0];
rocblas_int ldc = args[arg_num - 1].get_shape().strides()[dim_0]; rocblas_int ldc = args[arg_num - 1].get_shape().strides()[dim_0];
...@@ -121,7 +121,8 @@ argument miopen_quant_gemm::compute(context& ctx, ...@@ -121,7 +121,8 @@ argument miopen_quant_gemm::compute(context& ctx,
m, m,
k, k,
&alpha_r, &alpha_r,
(!transb) ? to_pointer(args[arg_num - 2]) : to_pointer(args.at(1)), (!transb) ? to_pointer(args[arg_num - 2])
: to_pointer(args.at(1)),
rocblas_datatype_i8_r, rocblas_datatype_i8_r,
ldb, ldb,
transa ? to_pointer(args[arg_num - 3]) : to_pointer(args.at(0)), transa ? to_pointer(args[arg_num - 3]) : to_pointer(args.at(0)),
......
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