From a309ce5e33b22030bcac348c63576187676abee3 Mon Sep 17 00:00:00 2001 From: Determinant Date: Mon, 1 Jun 2015 17:37:13 +0800 Subject: add expand_frm, rearrange_frm, scale_row --- matrix/cukernel.h | 3 +++ 1 file changed, 3 insertions(+) (limited to 'matrix/cukernel.h') 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 -- cgit v1.2.3