From 51ee28426b992b1421ded46721ade7a57b0a1896 Mon Sep 17 00:00:00 2001 From: Determinant Date: Thu, 28 May 2015 11:28:24 +0800 Subject: sigmoid layer is now working; clean up Makefile --- matrix/generic/cumatrix.c | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'matrix/generic') diff --git a/matrix/generic/cumatrix.c b/matrix/generic/cumatrix.c index 2b3b9d4..f24280c 100644 --- a/matrix/generic/cumatrix.c +++ b/matrix/generic/cumatrix.c @@ -98,9 +98,8 @@ static int nerv_matrix_(sigmoid)(lua_State *L) { Matrix *a = luaT_checkudata(L, 1, nerv_matrix_(tname)); Matrix *b = luaT_checkudata(L, 2, nerv_matrix_(tname)); CHECK_SAME_DIMENSION(a, b); - cudak_(cuda_sigmoid)(a, b); - luaT_pushudata(L, b, nerv_matrix_(tname)); - return 1; + cudak_(cuda_sigmoid)(b, a); + return 0; } static int nerv_matrix_(sigmoid_grad)(lua_State *L) { -- cgit v1.2.3