aboutsummaryrefslogtreecommitdiff
path: root/matrix/cukernel.h
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-06-01 17:37:13 +0800
committerDeterminant <[email protected]>2015-06-01 17:37:13 +0800
commita309ce5e33b22030bcac348c63576187676abee3 (patch)
treeb43714a53b8f78a12a52cb0ee88c6ed7be786cac /matrix/cukernel.h
parentab12a9583bdd39884fde9bc2444e6fd1bc5f518e (diff)
add expand_frm, rearrange_frm, scale_row
Diffstat (limited to 'matrix/cukernel.h')
-rw-r--r--matrix/cukernel.h3
1 files changed, 3 insertions, 0 deletions
diff --git a/matrix/cukernel.h b/matrix/cukernel.h
index 232699d..178b7d3 100644
--- a/matrix/cukernel.h
+++ b/matrix/cukernel.h
@@ -10,4 +10,7 @@ void cudak_(cuda_softmax_denominator)(const Matrix *a, const Matrix *max, Matrix
void cudak_(cuda_softmax_final)(const Matrix *a, const Matrix *max, const Matrix *deno, Matrix *b);
void cudak_(cuda_add_row)(const Matrix *a, Matrix *b, double beta);
void cudak_(cuda_fill)(Matrix *a, double val);
+void cudak_(cuda_expand_frm)(const Matrix *a, Matrix *b, int context);
+void cudak_(cuda_rearrange_frm)(const Matrix *a, Matrix *b, int step);
+void cudak_(cuda_scale_row)(const Matrix *a, Matrix *b);
#endif