From fc4c5a71053b837ed6143659a6e7b45792ed9e51 Mon Sep 17 00:00:00 2001 From: txh18 Date: Tue, 17 Nov 2015 21:58:05 +0800 Subject: added atomicAdd for select_linear update, however, the result still seems unreproducable, I changed select_linear layer update back to line-by-line --- nerv/layer/affine.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'nerv/layer/affine.lua') diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua index 3ba9408..6a541e8 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -5,7 +5,7 @@ local AffineLayer = nerv.class('nerv.AffineLayer', 'nerv.Layer') function MatrixParam:read(handle) self.trans = self.gconf.cumat_type.new_from_host( - nerv.MMatrixFloat.load(handle)) + self.gconf.mmat_type.load(handle)) end function MatrixParam:write(handle) -- cgit v1.2.3