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

clang format

parent 1898f87c
......@@ -175,22 +175,22 @@ struct miopen_apply
void add_quant_gemm_op()
{
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();
// 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());
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());
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);
return prog->replace_instruction(ins, miopen_quant_gemm{op}, refs);
......
......@@ -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 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;
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