diff --git a/src/operator/FCImpl_CUDA_kernels.cu b/src/operator/FCImpl_CUDA_kernels.cu index cf98aabf4b9275368bbdcd12b3a379b52a10681e..f263f150c93c6882dad5047f789bdc5103360a6f 100644 --- a/src/operator/FCImpl_CUDA_kernels.cu +++ b/src/operator/FCImpl_CUDA_kernels.cu @@ -25,7 +25,7 @@ void fc_forward_cuda_kernel(std::size_t nbInputs, std::size_t inChannels, std::s for (std::size_t in = 0; in < inChannels; ++in) { sum += input[batch * inChannels + in] * weights[out * inChannels + in]; } - output[batch * outChannels + out] = sum + (noBias ? 0 : bias[out]); + output[batch * outChannels + out] = sum + (noBias ? 0 : bias[out]); } } }