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

Formatting

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