From ff6cb5b6ab9490204683368fd4e70bbeebd2a49e Mon Sep 17 00:00:00 2001 From: Determinant Date: Wed, 17 Feb 2016 20:25:45 +0800 Subject: ... --- nerv/lib/matrix/cukernel.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'nerv/lib/matrix/cukernel.h') 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 -- cgit v1.2.3