Skip to content

Commit

Permalink
Fix compile error (apache#181)
Browse files Browse the repository at this point in the history
The Eval(...) function takes a pointer to Tensor as the first argument, not the Tensor itself.
  • Loading branch information
yllan authored and sxjscience committed Nov 23, 2016
1 parent 58276ff commit 6ff0804
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mshadow/dot_engine-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -811,7 +811,7 @@ struct DotEngine<SV, xpu, 2, 1, 1, true, false, DType> {
rhs.dptr_, 1, lhs.dptr_, 1, dst.dptr_, dst.stride_);
} else {
DotEngine<SV, xpu, 2, 2, 2, true, false,
DType>::Eval(dst, lhs.FlatTo2D(), rhs.FlatTo2D(), scale);
DType>::Eval(p_dst, lhs.FlatTo2D(), rhs.FlatTo2D(), scale);
}
}
};
Expand Down

0 comments on commit 6ff0804

Please sign in to comment.