diff options
author | Qi Liu <liuq901@163.com> | 2016-02-24 16:16:03 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-02-24 16:16:03 +0800 |
commit | a51498d2761714f4e034f036b84b4b89a88e9539 (patch) | |
tree | ff78fabd169c3c0346453c7005c84b176ac49ca6 /nerv/lib/matrix/generic/cumatrix.c | |
parent | 9642bd16922b288c81dee25f17373466ae6888c4 (diff) |
update LSTM layer
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.c')
-rw-r--r-- | nerv/lib/matrix/generic/cumatrix.c | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.c b/nerv/lib/matrix/generic/cumatrix.c index 7b70607..1c74866 100644 --- a/nerv/lib/matrix/generic/cumatrix.c +++ b/nerv/lib/matrix/generic/cumatrix.c @@ -494,6 +494,15 @@ void nerv_matrix_(prefixsum_row)(Matrix *a, const Matrix *b, Status *status) { NERV_SET_STATUS(status, NERV_NORMAL, 0); } +void nerv_matrix_(diagonalize)(Matrix *a, Status *status) { + if (a->nrow != a->ncol) + NERV_EXIT_STATUS(status, MAT_MISMATCH_DIM, 0); + PROFILE_START + cudak_(cuda_diagonalize)(a); + PROFILE_STOP + NERV_SET_STATUS(status, NERV_NORMAL, 0); +} + static void cuda_matrix_(free)(MATRIX_ELEM *ptr, Status *status) { CUDA_SAFE_SYNC_CALL(cudaFree(ptr), status); NERV_SET_STATUS(status, NERV_NORMAL, 0); |