diff options
| -rw-r--r-- | lib/THCUNN/generic/PReLU.cu | 6 | ||||
| -rw-r--r-- | lib/THCUNN/generic/SparseLinear.cu | 2 |
2 files changed, 4 insertions, 4 deletions
diff --git a/lib/THCUNN/generic/PReLU.cu b/lib/THCUNN/generic/PReLU.cu index 949e3d9..05271f8 100644 --- a/lib/THCUNN/generic/PReLU.cu +++ b/lib/THCUNN/generic/PReLU.cu @@ -135,7 +135,7 @@ void THNN_(PReLU_accGradParameters)( if (ndim == 2) { - THCTensor_(sum)(state, gradWeightBuf, gradInput, 0); + THCTensor_(sum)(state, gradWeightBuf, gradInput, 0, 1); THCTensor_(cadd)(state, gradWeight, gradWeight, scale, gradWeightBuf); } else @@ -147,8 +147,8 @@ void THNN_(PReLU_accGradParameters)( } THCTensor_(resize3d)(state, buffer, input->size[0], nOutputPlane, size3); THCTensor_(resize2d)(state, sumbuf, input->size[0], nOutputPlane); - THCTensor_(sum)(state, sumbuf, buffer, 2); - THCTensor_(sum)(state, gradWeightBuf, sumbuf, 0); + THCTensor_(sum)(state, sumbuf, buffer, 2, 1); + THCTensor_(sum)(state, gradWeightBuf, sumbuf, 0, 1); THCTensor_(cadd)(state, gradWeight, gradWeight, scale, gradWeightBuf); THCTensor_(free)(state, buffer); } diff --git a/lib/THCUNN/generic/SparseLinear.cu b/lib/THCUNN/generic/SparseLinear.cu index 70c9f5b..07eda62 100644 --- a/lib/THCUNN/generic/SparseLinear.cu +++ b/lib/THCUNN/generic/SparseLinear.cu @@ -206,7 +206,7 @@ void THNN_(SparseLinear_accGradParameters)( &one, THCTensor_(data)(state, gradWeight), inDim ); - THCTensor_(sum)(state, buf, gradOutput, 0); + THCTensor_(sum)(state, buf, gradOutput, 0, 1); THCTensor_(resize1d)(state, buf, outDim); THCTensor_(cadd)(state, gradBias, gradBias, scale, buf); |