Commit 04ba72c7 authored by Jing Zhang's avatar Jing Zhang
Browse files

formatting

parent 5a3c2297
...@@ -76,8 +76,11 @@ struct ReferenceGemmTranspose : public device::BaseOperator ...@@ -76,8 +76,11 @@ struct ReferenceGemmTranspose : public device::BaseOperator
arg.c_m0_m1_n0_n1_(m0, m1, n0, n1) = v_c; arg.c_m0_m1_n0_n1_(m0, m1, n0, n1) = v_c;
}; };
make_ParallelTensorFunctor( make_ParallelTensorFunctor(f_mk_kn_m0m1n0n1,
f_mk_kn_m0m1n0n1, arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[0], arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[1], arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[2],arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[3])( arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[0],
arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[1],
arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[2],
arg.c_m0_m1_n0_n1_.mDesc.GetLengths()[3])(
std::thread::hardware_concurrency()); std::thread::hardware_concurrency());
return 0; return 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