aboutsummaryrefslogtreecommitdiff
path: root/nerv/lib/matrix/cukernel.h
diff options
context:
space:
mode:
authorDeterminant <ted.sybil@gmail.com>2016-02-17 20:25:45 +0800
committerDeterminant <ted.sybil@gmail.com>2016-02-17 20:25:45 +0800
commitff6cb5b6ab9490204683368fd4e70bbeebd2a49e (patch)
treeb115697ad66c4ca82d705847ba3a2c03132126ea /nerv/lib/matrix/cukernel.h
parent5b655d222e45cd9b956b62272715ecbc39025e8c (diff)
...tnn
Diffstat (limited to 'nerv/lib/matrix/cukernel.h')
-rw-r--r--nerv/lib/matrix/cukernel.h2
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