Commit d18d38c4 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 1898f87c
...@@ -175,22 +175,22 @@ struct miopen_apply ...@@ -175,22 +175,22 @@ struct miopen_apply
void add_quant_gemm_op() void add_quant_gemm_op()
{ {
apply_map.emplace("quant_dot", [=](instruction_ref ins) { apply_map.emplace("quant_dot", [=](instruction_ref ins) {
auto&& op = any_cast<op::quant_dot>(ins->get_operator()); auto&& op = any_cast<op::quant_dot>(ins->get_operator());
std::vector<instruction_ref> refs = ins->inputs(); std::vector<instruction_ref> refs = ins->inputs();
// add additional arguments if need packing // add additional arguments if need packing
if (refs.at(0)->get_shape().transposed()) if(refs.at(0)->get_shape().transposed())
{ {
auto pack_a = insert_allocation(refs.at(0), refs.at(0)->get_shape()); auto pack_a = insert_allocation(refs.at(0), refs.at(0)->get_shape());
refs.push_back(pack_a); refs.push_back(pack_a);
} }
if (!refs.at(1)->get_shape().transposed()) if(!refs.at(1)->get_shape().transposed())
{ {
auto pack_b = insert_allocation(refs.at(1), refs.at(1)->get_shape()); auto pack_b = insert_allocation(refs.at(1), refs.at(1)->get_shape());
refs.push_back(pack_b); refs.push_back(pack_b);
} }
auto output = insert_allocation(ins, ins->get_shape()); auto output = insert_allocation(ins, ins->get_shape());
refs.push_back(output); refs.push_back(output);
return prog->replace_instruction(ins, miopen_quant_gemm{op}, refs); return prog->replace_instruction(ins, miopen_quant_gemm{op}, refs);
......
...@@ -84,7 +84,7 @@ argument miopen_quant_gemm::compute(context& ctx, ...@@ -84,7 +84,7 @@ argument miopen_quant_gemm::compute(context& ctx,
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[2].get_shape().strides()[dim_0]; rocblas_int ldc = args[2].get_shape().strides()[dim_0];
auto arg_b = args.at(1); auto arg_b = args.at(1);
std::size_t pack_arg_num = 0; std::size_t pack_arg_num = 0;
if(!transb) if(!transb)
{ {
......
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