Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix softmax sum kernel

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-08 22:09:18 +02:00
parent 7bc329a043
commit e419a93408
5 changed files with 49 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/kernels/activation_functions.cuh
View File

@@ -51,8 +51,7 @@ __global__ void softmax_exp(
*/
__global__ void softmax_sum(
const float* __restrict__ d_vector,
float* __restrict__ d_output,
const unsigned int w
float* __restrict__ d_output
);
/**