casadi_mtimes_dense_sparse.hpp
1 //
2 // MIT No Attribution
3 //
4 // Copyright (C) 2010-2023 Joel Andersson, Joris Gillis, Moritz Diehl, KU Leuven.
5 //
6 // Permission is hereby granted, free of charge, to any person obtaining a copy of this
7 // software and associated documentation files (the "Software"), to deal in the Software
8 // without restriction, including without limitation the rights to use, copy, modify,
9 // merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
10 // permit persons to whom the Software is furnished to do so.
11 //
12 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
13 // INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
14 // PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
15 // HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
16 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
17 // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
18 //
19 
20 // SYMBOL "mtimes_dense_sparse"
21 // z (m x p, dense col-major) += x (m x n, dense col-major) * y (n x p, CCS sparse)
22 template<typename T1>
23 void casadi_mtimes_dense_sparse(const T1* x, casadi_int nrow_x,
24  const T1* y, const casadi_int* sp_y, T1* z) {
25  casadi_int j, kk, i, ncol_y;
26  const casadi_int *colind_y, *row_y;
27  ncol_y = sp_y[1];
28  colind_y = sp_y + 2;
29  row_y = sp_y + 2 + ncol_y + 1;
30  for (j = 0; j < ncol_y; ++j) {
31  T1* zj = z + j * nrow_x;
32  for (kk = colind_y[j]; kk < colind_y[j + 1]; ++kk) {
33  casadi_int rr = row_y[kk];
34  T1 yk = y[kk];
35  const T1* x_col = x + rr * nrow_x;
36  for (i = 0; i < nrow_x; ++i) zj[i] += x_col[i] * yk;
37  }
38  }
39 }