diff options
author | TianxingHe <htx_2006@hotmail.com> | 2016-01-19 20:13:49 -0800 |
---|---|---|
committer | TianxingHe <htx_2006@hotmail.com> | 2016-01-19 20:13:49 -0800 |
commit | dcad8a3f80fc55ca93984d981f9b829d2e4ea728 (patch) | |
tree | 61b9bc1d043883bb5d85dcb86cfb621396d75c41 /nerv/lib/matrix/generic/cumatrix.h | |
parent | 7449dd19c4d1669b483693f61add9d574e46f0b2 (diff) | |
parent | 37dec2610c92d03813c4e91ed58791ab60da6646 (diff) |
Merge pull request #21 from cloudygoose/txh18/rnnlm
Txh18/rnnlm new changes to lm side
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
-rw-r--r-- | nerv/lib/matrix/generic/cumatrix.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h index aa8805a..e82dccd 100644 --- a/nerv/lib/matrix/generic/cumatrix.h +++ b/nerv/lib/matrix/generic/cumatrix.h @@ -45,7 +45,13 @@ void nerv_matrix_(copy_rows_fromh_by_idx)(Matrix *a, const Matrix *b, const Matrix *idx, int b_begin, Status *status); void nerv_matrix_(copy_rows_fromd_by_idx)(Matrix *a, const Matrix *b, const Matrix *idx, int b_begin, Status *status); -void nerv_matrix_(update_select_rows)(Matrix *c, const Matrix *a, const Matrix *idx, double alpha, double beta, Status *status); +void nerv_matrix_(copy_rows_fromd_by_colidx)(Matrix *a, const Matrix *b, + const Matrix *idx, int b_begin, Status *status); + +#ifdef __NERV_FUTURE_CUDA_7 +void nerv_matrix_(update_select_rows_by_rowidx)(Matrix *c, const Matrix *a, const Matrix *idx, double alpha, double beta, Status *status); +void nerv_matrix_(update_select_rows_by_colidx)(Matrix *c, const Matrix *a, const Matrix *idx, double alpha, double beta, Status *status); +#endif void nerv_matrix_(expand_frm)(Matrix *a, const Matrix *b, int context, Status *status); |