Skip to content

Commit

Permalink
Revert the storage order with some comments
Browse files Browse the repository at this point in the history
  • Loading branch information
tomdol committed Nov 27, 2024
1 parent 1c0de4f commit 09670cf
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions compute/cker/include/cker/operation/optimized/BatchMatMul.h
Original file line number Diff line number Diff line change
Expand Up @@ -33,14 +33,17 @@ inline void BatchMatMul(const BatchMatMulParams &params, const float *lhs_data,
const float *rhs_data, float *output_data)
{
MatrixParams<float> lhs_params;
lhs_params.order = Order::kRowMajor; // ignored by GemmImplUsingEigen
lhs_params.rows = params.lhs_rows;
lhs_params.cols = params.lhs_cols;

MatrixParams<float> rhs_params;
lhs_params.order = Order::kRowMajor; // ignored by GemmImplUsingEigen
rhs_params.rows = params.rhs_rows;
rhs_params.cols = params.rhs_cols;

MatrixParams<float> dst_params;
lhs_params.order = Order::kRowMajor; // ignored by GemmImplUsingEigen
dst_params.rows = params.lhs_rows;
dst_params.cols = params.rhs_cols;

Expand Down

0 comments on commit 09670cf

Please sign in to comment.