Commit 49e7cd87 authored by Paul's avatar Paul
Browse files

Formatting

parent 7c885dbb
...@@ -156,9 +156,10 @@ struct miopen_gemm ...@@ -156,9 +156,10 @@ struct miopen_gemm
visit_all(result, from_gpu(args[1]), from_gpu(args[2]))( visit_all(result, from_gpu(args[1]), from_gpu(args[2]))(
[&](auto output, auto input1, auto input2) { [&](auto output, auto input1, auto input2) {
dfor(input1.get_shape().lens()[0], input2.get_shape().lens()[1], input2.get_shape().lens()[0])([&](auto i, auto j, auto k) { dfor(input1.get_shape().lens()[0],
output(i, j) += input1(i, k) * input2(k, j); input2.get_shape().lens()[1],
}); input2.get_shape().lens()[0])(
[&](auto i, auto j, auto k) { output(i, j) += input1(i, k) * input2(k, j); });
}); });
return to_gpu(result); return to_gpu(result);
} }
......
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