diff options
| author | Determinant <ted.sybil@gmail.com> | 2016-02-17 20:14:06 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2016-02-17 20:14:06 +0800 |
| commit | 0ee43c21af4fcd3aed070b1f5ad1eb9feb2ad159 (patch) | |
| tree | ceb1d38328767fb657bc0d37ec6e513b08a86277 /nerv/lib/matrix/generic/cumatrix.c | |
| parent | 490a10c2130773bd022f05513fa2905b6a6c6e91 (diff) | |
try to merge manually
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.c')
| -rw-r--r-- | nerv/lib/matrix/generic/cumatrix.c | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.c b/nerv/lib/matrix/generic/cumatrix.c index bf93b77..7b70607 100644 --- a/nerv/lib/matrix/generic/cumatrix.c +++ b/nerv/lib/matrix/generic/cumatrix.c @@ -486,6 +486,14 @@ void nerv_matrix_(scale_rows_by_row)(Matrix *a, const Matrix *b, NERV_SET_STATUS(status, NERV_NORMAL, 0); } +void nerv_matrix_(prefixsum_row)(Matrix *a, const Matrix *b, Status *status) { + CHECK_SAME_DIMENSION(a, b, status); + PROFILE_START + cudak_(cuda_prefixsum_row)(b, 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); |