diff options
Diffstat (limited to 'lib/THCUNN/generic/SparseLinear.cu')
-rw-r--r-- | lib/THCUNN/generic/SparseLinear.cu | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/lib/THCUNN/generic/SparseLinear.cu b/lib/THCUNN/generic/SparseLinear.cu index 6838cac..f22b233 100644 --- a/lib/THCUNN/generic/SparseLinear.cu +++ b/lib/THCUNN/generic/SparseLinear.cu @@ -127,8 +127,8 @@ void THNN_(SparseLinear_accGradParameters)( THCTensor *gradBias, THCTensor *weight, THCTensor *bias, - accreal weightDecay, - accreal scale) + double weightDecay, + double scale) { long outDim = THCTensor_(size)(state, weight, 0); long inDim = THCTensor_(size)(state, weight, 1); @@ -237,8 +237,8 @@ void THNN_(SparseLinear_legacyAccGradParameters)( THCTensor *gradBias, THCTensor *weight, THCTensor *bias, - accreal weightDecay, - accreal scale) { + double weightDecay, + double scale) { THError("CUDA does not support legacy input format, please use a table of nnz x 2 vectors"); } @@ -259,7 +259,7 @@ void THNN_(SparseLinear_updateParameters)( THCTensor *gradWeight, THCTensor *gradBias, THCTensor *lastInput, - accreal learningRate) { + double learningRate) { THCTensor_(cadd)(state, weight, weight, -learningRate, gradWeight); THCTensor_(cadd)(state, bias, bias, -learningRate, gradBias); } |