"src/targets/gpu/vscode:/vscode.git/clone" did not exist on "4e3ca5861a59c7e82aad43d49ec3a622188d96d8"
Commit 0f2b0684 authored by Paul's avatar Paul
Browse files

Formatting

parent cff83eda
......@@ -227,10 +227,10 @@ struct cpu_gemm
std::string name() const { return "cpu::gemm"; }
shape compute_shape(std::vector<shape> inputs) const { return op.compute_shape(inputs); }
template<class T>
template <class T>
using matrix = blaze::CustomMatrix<T, blaze::unaligned, blaze::unpadded>; // NOLINT
template<class T>
template <class T>
static auto make_mat(tensor_view<T> x)
{
const auto& s = x.get_shape();
......@@ -241,7 +241,7 @@ struct cpu_gemm
return matrix<T>{x.data(), s.lens()[0], s.lens()[1]};
}
template<class T, class F>
template <class T, class F>
static void visit_mat(tensor_view<T> x, F f)
{
auto mat = make_mat(x);
......@@ -258,7 +258,7 @@ struct cpu_gemm
visit_mat(amat, [&](const auto& a) {
visit_mat(bmat, [&](const auto& b) {
auto c = make_mat(cmat);
if (op.alpha == 1.0 and op.beta == 0.0)
if(op.alpha == 1.0 and op.beta == 0.0)
c = a * b;
else
c = (a * b) * op.alpha + op.beta * c;
......
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