aboutsummaryrefslogtreecommitdiff
path: root/nerv/lib/matrix/cukernel.h
diff options
context:
space:
mode:
authorDeterminant <ted.sybil@gmail.com>2016-02-15 16:18:13 +0800
committerDeterminant <ted.sybil@gmail.com>2016-02-15 16:18:13 +0800
commita7608a67f16f745309409f6a004354121e0b3ba6 (patch)
treeb36ec7c4e8b252f540f049d374a13915ca5ba9a3 /nerv/lib/matrix/cukernel.h
parent3362020a6bc43766a92882abe6d127c8bb98a628 (diff)
move curand_gen out of the matrix layout
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 39d42eb..04d0e28 100644
--- a/nerv/lib/matrix/cukernel.h
+++ b/nerv/lib/matrix/cukernel.h
@@ -25,4 +25,6 @@ void cudak_(cuda_scale_rows_by_col)(const Matrix *a, Matrix *b);
void cudak_(cuda_decompress)(const Matrix *a, Matrix *b);
void cudak_(cuda_copy_rows_by_idx)(const Matrix *a, Matrix *b,
const Matrix *idx, int b_begin);
+void cudak_(cuda_copy_rows_by_colidx)(const Matrix *a, Matrix *b,
+ const Matrix *idx, int b_begin);
#endif