aboutsummaryrefslogtreecommitdiff
path: root/nerv/lib/matrix/generic/cumatrix.h
diff options
context:
space:
mode:
authorDeterminant <ted.sybil@gmail.com>2015-08-28 13:21:52 +0800
committerDeterminant <ted.sybil@gmail.com>2015-08-28 13:21:52 +0800
commit1a9f63e351582f54fec7817927168cb1dbb0c1d6 (patch)
treec340b648c60d93b956be5956fa03233383e37e5d /nerv/lib/matrix/generic/cumatrix.h
parent8bf9c7575ffeeabb3924e9e02a35afe187071fe2 (diff)
support gpu buffering
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
-rw-r--r--nerv/lib/matrix/generic/cumatrix.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h
index 3f1f8a3..04e8c5a 100644
--- a/nerv/lib/matrix/generic/cumatrix.h
+++ b/nerv/lib/matrix/generic/cumatrix.h
@@ -40,6 +40,8 @@ void nerv_matrix_(log_elem)(Matrix *b, const Matrix *a, Status *status);
Matrix *nerv_matrix_(decompress)(const Matrix *a, int orig_col, Status *status);
void nerv_matrix_(copy_rows_fromh_by_idx)(Matrix *a, const Matrix *b,
const Matrix *idx, int b_begin, Status *status);
+void nerv_matrix_(copy_rows_fromd_by_idx)(Matrix *a, const Matrix *b,
+ const Matrix *idx, int b_begin, Status *status);
void nerv_matrix_(expand_frm)(Matrix *a, const Matrix *b,
int context, Status *status);