diff options
Diffstat (limited to 'nerv/lib/matrix/cukernel.h')
-rw-r--r-- | nerv/lib/matrix/cukernel.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/lib/matrix/cukernel.h b/nerv/lib/matrix/cukernel.h index 4654d29..c84200e 100644 --- a/nerv/lib/matrix/cukernel.h +++ b/nerv/lib/matrix/cukernel.h @@ -34,4 +34,6 @@ void cudak_(cuda_copy_rows_by_colidx)(const Matrix *a, Matrix *b, double alpha, double beta); #endif +void cudak_(cuda_prefixsum_row)(const Matrix *a, Matrix *b); + #endif |